diff --git a/comfy/supported_models.py b/comfy/supported_models.py index 776091d42..00941da53 100644 --- a/comfy/supported_models.py +++ b/comfy/supported_models.py @@ -845,6 +845,8 @@ class Lens(supported_models_base.BASE): unet_extra_config = {} latent_format = latent_formats.Flux2 + memory_usage_factor = 4.0 + supported_inference_dtypes = [torch.bfloat16, torch.float32] # fp16 causes NaNs vae_key_prefix = ["vae."]