diff options
| author | Muhammad Rizqi Nur <rizqinur2010@gmail.com> | 2022-11-17 11:04:10 +0000 |
|---|---|---|
| committer | Muhammad Rizqi Nur <rizqinur2010@gmail.com> | 2022-11-19 04:44:37 +0000 |
| commit | 9fdc343dcaee70f1a0ff15c0cc668dbd487abc61 (patch) | |
| tree | e19995b6e4d8b0055cf7dbf30316baf8ab75fca8 /modules/sd_vae.py | |
| parent | c7be83bf0240498d9382e2afeaa3f0677d26c7f6 (diff) | |
| download | stable-diffusion-webui-gfx803-9fdc343dcaee70f1a0ff15c0cc668dbd487abc61.tar.gz stable-diffusion-webui-gfx803-9fdc343dcaee70f1a0ff15c0cc668dbd487abc61.tar.bz2 stable-diffusion-webui-gfx803-9fdc343dcaee70f1a0ff15c0cc668dbd487abc61.zip | |
Fix model caching requiring deepcopy
Diffstat (limited to 'modules/sd_vae.py')
| -rw-r--r-- | modules/sd_vae.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/sd_vae.py b/modules/sd_vae.py index 13bf3d31..5b4709b5 100644 --- a/modules/sd_vae.py +++ b/modules/sd_vae.py @@ -4,6 +4,7 @@ from collections import namedtuple from modules import shared, devices, script_callbacks from modules.paths import models_path import glob +from copy import deepcopy model_dir = "Stable-diffusion" @@ -40,7 +41,7 @@ def store_base_vae(model): global base_vae, checkpoint_info if checkpoint_info != model.sd_checkpoint_info: assert not loaded_vae_file, "Trying to store non-base VAE!" - base_vae = model.first_stage_model.state_dict().copy() + base_vae = deepcopy(model.first_stage_model.state_dict()) checkpoint_info = model.sd_checkpoint_info |