Add memory usage factor for lens model. (#14124)

2026-05-27 17:37:39 +08:00 · 2026-05-26 18:06:51 -07:00 · 2026-05-26 18:06:51 -07:00 · e75a92c1b6
commit e75a92c1b6
parent d8d860a588
1 changed files with 2 additions and 0 deletions
--- 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."]