diff options
| author | Uminosachi <49424133+Uminosachi@users.noreply.github.com> | 2023-08-20 23:06:51 +0900 |
|---|---|---|
| committer | Uminosachi <49424133+Uminosachi@users.noreply.github.com> | 2023-08-20 23:06:51 +0900 |
| commit | 549b0fc5267e9539f321f0891aa757619b7248cb (patch) | |
| tree | 4ca368bd87110d770ab5ebbed76d1e3a9eda149e | |
| parent | af5d2e8e5fc4440691fb7f1aa3492def1c755722 (diff) | |
Change where VAE state are stored in model
| -rw-r--r-- | modules/sd_models.py | 2 | ||||
| -rw-r--r-- | modules/sd_vae.py | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index dd749122..d3775ec6 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -498,8 +498,6 @@ class SdModelData: pass
if v is not None:
- v.base_vae = sd_vae.base_vae
- v.loaded_vae_file = sd_vae.loaded_vae_file
self.loaded_sd_models.insert(0, v)
diff --git a/modules/sd_vae.py b/modules/sd_vae.py index dbade067..ee118656 100644 --- a/modules/sd_vae.py +++ b/modules/sd_vae.py @@ -192,7 +192,7 @@ def load_vae_dict(filename, map_location): def load_vae(model, vae_file=None, vae_source="from unknown source"): - global vae_dict, loaded_vae_file + global vae_dict, base_vae, loaded_vae_file # save_settings = False cache_enabled = shared.opts.sd_vae_checkpoint_cache > 0 @@ -230,6 +230,8 @@ def load_vae(model, vae_file=None, vae_source="from unknown source"): restore_base_vae(model) loaded_vae_file = vae_file + model.base_vae = base_vae + model.loaded_vae_file = loaded_vae_file # don't call this from outside |