From 4fe1dd6a1cad5aba622ab98c50ea0276b9b12890 Mon Sep 17 00:00:00 2001 From: Hacker 17082006 Date: Sat, 14 Jan 2023 14:59:29 +0700 Subject: [PATCH] Wrong indention --- library/model_util.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/library/model_util.py b/library/model_util.py index 6bbc3417..7d16338d 100644 --- a/library/model_util.py +++ b/library/model_util.py @@ -1106,12 +1106,12 @@ def load_vae(vae_id, dtype): if vae_key.startswith(VAE_PREFIX): full_model = True break - if not full_model: - sd = {} - for key, value in vae_sd.items(): - sd[VAE_PREFIX + key] = value - vae_sd = sd - del sd + if not full_model: + sd = {} + for key, value in vae_sd.items(): + sd[VAE_PREFIX + key] = value + vae_sd = sd + del sd converted_vae_checkpoint = convert_ldm_vae_checkpoint(vae_sd, vae_config) vae = AutoencoderKL(**vae_config)