From 5b7f873fada4b13c18cee06143e92c51e93a0fa5 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Sat, 19 Aug 2023 22:07:10 +0300 Subject: [PATCH] Fix sequential offloading --- modules/processing_diffusers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/processing_diffusers.py b/modules/processing_diffusers.py index 652a0d167..54a709899 100644 --- a/modules/processing_diffusers.py +++ b/modules/processing_diffusers.py @@ -60,7 +60,8 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro unet_device = model.unet.device model.unet.to(devices.cpu) devices.torch_gc() - model.vae.to(devices.device) + if not shared.cmd_opts.lowvram and not shared.opts.diffusers_seq_cpu_offload: + model.vae.to(devices.device) latents.to(model.vae.device) decoded = model.vae.decode(latents / model.vae.config.scaling_factor, return_dict=False)[0] if shared.opts.diffusers_move_unet and not model.has_accelerate: