diff options
| author | Muhammad Rizqi Nur <rizqinur2010@gmail.com> | 2022-11-02 05:51:46 +0000 |
|---|---|---|
| committer | Muhammad Rizqi Nur <rizqinur2010@gmail.com> | 2022-11-02 05:51:46 +0000 |
| commit | 056f06d3738c267b1014e6e8e1ef5bd97af1fb45 (patch) | |
| tree | c80870ce44039b839b5c40cbe832794eecdba671 /webui.py | |
| parent | f8c6468d42e1202f7aeaeb961ab003aa0a2daf99 (diff) | |
| download | stable-diffusion-webui-gfx803-056f06d3738c267b1014e6e8e1ef5bd97af1fb45.tar.gz stable-diffusion-webui-gfx803-056f06d3738c267b1014e6e8e1ef5bd97af1fb45.tar.bz2 stable-diffusion-webui-gfx803-056f06d3738c267b1014e6e8e1ef5bd97af1fb45.zip | |
Reload VAE without reloading sd checkpoint
Diffstat (limited to 'webui.py')
| -rw-r--r-- | webui.py | 4 |
1 files changed, 1 insertions, 3 deletions
@@ -81,9 +81,7 @@ def initialize(): modules.sd_vae.refresh_vae_list()
modules.sd_models.load_model()
shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights()))
- # I don't know what needs to be done to only reload VAE, with all those hijacks callbacks, and lowvram,
- # so for now this reloads the whole model too
- shared.opts.onchange("sd_vae", wrap_queued_call(lambda: modules.sd_models.reload_model_weights(force=True)), call=False)
+ shared.opts.onchange("sd_vae", wrap_queued_call(lambda: modules.sd_vae.reload_vae_weights()), call=False)
shared.opts.onchange("sd_hypernetwork", wrap_queued_call(lambda: modules.hypernetworks.hypernetwork.load_hypernetwork(shared.opts.sd_hypernetwork)))
shared.opts.onchange("sd_hypernetwork_strength", modules.hypernetworks.hypernetwork.apply_strength)
|