Add torch_gc to state.nextjob, vae and upscale

2026-09-20 01:31:13 +02:00 · 2024-06-20 14:47:30 +03:00
parent eb6e3c337c
commit 092a326c09
4 changed files with 11 additions and 3 deletions
@@ -400,6 +400,7 @@ def resize_hires(p, latents): # input=latents output=pil if not latent_upscaler
        else:
            resized_image = img
        resized_images.append(resized_image)
+    devices.torch_gc()
    return resized_images