From 1ec9f708e5cbb7a2d74fa35f56735d80991163c2 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 7 Dec 2025 11:49:22 -0800
Subject: [PATCH] Clear the torch cache between sequential image generations

---
 modules/ui_image_generation.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py
index 6b39c5b5..bcc5bb2d 100644
--- a/modules/ui_image_generation.py
+++ b/modules/ui_image_generation.py
@@ -848,6 +848,9 @@ def generate(state, save_images=True):
             if shared.stop_everything:
                 break
 
+            if batch_idx > 0:
+                clear_torch_cache()
+
             generator.manual_seed(int(seed + batch_idx))
 
             # Generate prompt variation if enabled