From 31e1bb01ffb0a905229e582ac713f61693c2e0db Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 28 Jul 2023 23:23:36 +0300 Subject: [PATCH] Fix VAE reloading --- modules/sd_vae.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/sd_vae.py b/modules/sd_vae.py index d0a2ec847..c2356d3cd 100644 --- a/modules/sd_vae.py +++ b/modules/sd_vae.py @@ -227,7 +227,7 @@ def reload_vae_weights(sd_model=None, vae_file=unspecified): return if shared.cmd_opts.lowvram or shared.cmd_opts.medvram: lowvram.send_everything_to_cpu() - else: + elif shared.backend == shared.Backend.ORIGINAL or not shared.opts.diffusers_seq_cpu_offload: sd_model.to(devices.cpu) if shared.backend == shared.Backend.ORIGINAL: @@ -238,7 +238,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 shared.cmd_opts.lowvram and not shared.cmd_opts.medvram and (shared.backend == shared.Backend.ORIGINAL or not shared.opts.diffusers_seq_cpu_offload): sd_model.to(devices.device) shared.log.info(f"VAE weights loaded: {vae_file}") return sd_model