aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_models.py
diff options
context:
space:
mode:
authorMuhammad Rizqi Nur <rizqinur2010@gmail.com>2022-11-03 04:10:53 +0000
committerMuhammad Rizqi Nur <rizqinur2010@gmail.com>2022-11-19 04:41:41 +0000
commitabc1e79a5da24a1ea0f4bceedcdf225f32010aa8 (patch)
treed6a840a8d7af4c5fdc86e9bff53cb844be98c9ea /modules/sd_models.py
parent8ab4927452b04dcd30847eaf92ea7a9f3b9c74e1 (diff)
downloadstable-diffusion-webui-gfx803-abc1e79a5da24a1ea0f4bceedcdf225f32010aa8.tar.gz
stable-diffusion-webui-gfx803-abc1e79a5da24a1ea0f4bceedcdf225f32010aa8.tar.bz2
stable-diffusion-webui-gfx803-abc1e79a5da24a1ea0f4bceedcdf225f32010aa8.zip
Fix base VAE caching was done after loading VAE, also add safeguard
Diffstat (limited to 'modules/sd_models.py')
-rw-r--r--modules/sd_models.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py
index 80addf03..e4dba62c 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -220,6 +220,7 @@ def load_model_weights(model, checkpoint_info, vae_file="auto"):
model.sd_model_checkpoint = checkpoint_file
model.sd_checkpoint_info = checkpoint_info
+ sd_vae.clear_loaded_vae()
sd_vae.load_vae(model, vae_file)