Put VAE key name in model config.

2025-09-14 21:45:06 +00:00 · 2024-01-30 02:24:38 -05:00
parent ed2fa105ae
commit da7a8df0d2
2 changed files with 2 additions and 1 deletions
--- a/comfy/supported_models_base.py
+++ b/comfy/supported_models_base.py
@@ -21,6 +21,7 @@ class BASE:
    noise_aug_config = None
    sampling_settings = {}
    latent_format = latent_formats.LatentFormat
+    vae_key_prefix = ["first_stage_model."]

    manual_cast_dtype = None