From a4d59d175dc382af65d4348f8cff87e8df6e0ddc Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Wed, 11 Sep 2024 08:27:58 -0400 Subject: [PATCH] cleanup --- modules/processing_vae.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/processing_vae.py b/modules/processing_vae.py index a0acb6e90..8e9063b89 100644 --- a/modules/processing_vae.py +++ b/modules/processing_vae.py @@ -35,6 +35,7 @@ def create_latents(image, p, dtype=None, device=None): def full_vae_decode(latents, model): t0 = time.time() if debug: + devices.torch_gc(force=True) shared.mem_mon.reset() base_device = None if shared.opts.diffusers_move_unet and not getattr(model, 'has_accelerate', False):