i

2026-04-30 03:01:15 +00:00 · 2024-01-30 21:58:16 -08:00
parent 93cc1b9363
commit 862b61830c
2 changed files with 2 additions and 1 deletions
--- a/ldm_patched/modules/supported_models_base.py
+++ b/ldm_patched/modules/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