Fix sequential offloading

Symbiomatrix · Aug 19, 2023 · 5b7f873 · 5b7f873
1 parent 87bb354
commit 5b7f873
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/processing_diffusers.py b/modules/processing_diffusers.py
@@ -60,7 +60,8 @@ def full_vae_decode(latents, model):
             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: