diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-08 18:49:47 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-08 18:49:47 +0000 |
| commit | 54c3e5c913b17622bed4ff4d03df488b80611e21 (patch) | |
| tree | 94f91922b3d0e6142837f6873ac9c12ba79d3f61 /modules/sd_models.py | |
| parent | 5a0db84b6c7322082c7532df11a29a95a59a612b (diff) | |
| parent | 70c63c1208d33bf02e15c4e310bac83f12ee8625 (diff) | |
| download | stable-diffusion-webui-gfx803-54c3e5c913b17622bed4ff4d03df488b80611e21.tar.gz stable-diffusion-webui-gfx803-54c3e5c913b17622bed4ff4d03df488b80611e21.tar.bz2 stable-diffusion-webui-gfx803-54c3e5c913b17622bed4ff4d03df488b80611e21.zip | |
Merge branch 'dev' into refiner
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 981aa93d..a97af215 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -372,7 +372,7 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer sd_vae.delete_base_vae()
sd_vae.clear_loaded_vae()
- vae_file, vae_source = sd_vae.resolve_vae(checkpoint_info.filename)
+ vae_file, vae_source = sd_vae.resolve_vae(checkpoint_info.filename).tuple()
sd_vae.load_vae(model, vae_file, vae_source)
timer.record("load VAE")
@@ -715,6 +715,7 @@ def reload_model_weights(sd_model=None, info=None): print(f"Weights loaded in {timer.summary()}.")
model_data.set_sd_model(sd_model)
+ sd_unet.apply_unet()
return sd_model
|