diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-22 15:49:08 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-22 15:49:08 +0000 |
| commit | 016554e43740e0b7ded75e89255de81270de9d6c (patch) | |
| tree | 127ecb9d92ce53f8e351270e8ef1c27784af3088 /modules/sd_vae.py | |
| parent | bb7dd7b64668d4b645dba38a3bc52be452d14eb8 (diff) | |
| download | stable-diffusion-webui-gfx803-016554e43740e0b7ded75e89255de81270de9d6c.tar.gz stable-diffusion-webui-gfx803-016554e43740e0b7ded75e89255de81270de9d6c.tar.bz2 stable-diffusion-webui-gfx803-016554e43740e0b7ded75e89255de81270de9d6c.zip | |
add --medvram-sdxl
Diffstat (limited to 'modules/sd_vae.py')
| -rw-r--r-- | modules/sd_vae.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_vae.py b/modules/sd_vae.py index ee118656..669097da 100644 --- a/modules/sd_vae.py +++ b/modules/sd_vae.py @@ -263,7 +263,7 @@ def reload_vae_weights(sd_model=None, vae_file=unspecified): if loaded_vae_file == vae_file: return - if shared.cmd_opts.lowvram or shared.cmd_opts.medvram: + if sd_model.lowvram: lowvram.send_everything_to_cpu() else: sd_model.to(devices.cpu) @@ -275,7 +275,7 @@ def reload_vae_weights(sd_model=None, vae_file=unspecified): sd_hijack.model_hijack.hijack(sd_model) script_callbacks.model_loaded_callback(sd_model) - if not shared.cmd_opts.lowvram and not shared.cmd_opts.medvram: + if not sd_model.lowvram: sd_model.to(devices.device) print("VAE weights loaded.") |