diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-21 07:10:19 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-08-21 07:10:19 +0300 |
commit | 5a3fe7a8d1c554982fb3226bda3ee6c4b10bb56c (patch) | |
tree | 01aae591c92fa1e4f0b57a488cc30fc09f2b63a8 /modules/sd_vae.py | |
parent | 42b72fe2463bc06a97935bc7a7770a9d562269d8 (diff) | |
parent | be301f224d26ac4363ce3bd8bcb510b00bd6db27 (diff) |
Merge pull request #12685 from Uminosachi/fix-vae-mismatch
Fix SD VAE switch error after model reuse
Diffstat (limited to 'modules/sd_vae.py')
-rw-r--r-- | modules/sd_vae.py | 4 |
1 files changed, 3 insertions, 1 deletions
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 |