diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-12-10 09:58:20 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-10 09:58:20 +0300 |
| commit | ec5e072124a553db7b5247a8745924faae6070c2 (patch) | |
| tree | 5821bdf7c426ada74e7cb58b1bd3274bc70b2e8f /modules/sd_models.py | |
| parent | e11d0d43b1048c23b31b894baae6352d057332c6 (diff) | |
| parent | 8662b5e57fb66ca16237107e67af97a709457eab (diff) | |
Merge pull request #4841 from R-N/vae-fix-none
Fix None option of VAE selector
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 139952ba..5b37f3fe 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -227,6 +227,8 @@ def load_model_weights(model, checkpoint_info, vae_file="auto"): model.sd_model_checkpoint = checkpoint_file
model.sd_checkpoint_info = checkpoint_info
+ sd_vae.delete_base_vae()
+ sd_vae.clear_loaded_vae()
vae_file = sd_vae.resolve_vae(checkpoint_file, vae_file=vae_file)
sd_vae.load_vae(model, vae_file)
|