hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix base VAE caching was done after loading VAE, also add safeguard

Browse Source
This commit is contained in:
Muhammad Rizqi Nur
2022-11-03 11:10:53 +07:00
parent 8ab4927452
commit abc1e79a5d
2 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/sd_models.py
View File

@@ -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)