From e7a15a0361013250340e4ceefe52319fac851c3b Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 2 Dec 2025 06:41:08 -0800
Subject: [PATCH] Clear the torch cache before generating

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

diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py
index 54cf4f31..92c447c8 100644
--- a/modules/ui_image_generation.py
+++ b/modules/ui_image_generation.py
@@ -667,6 +667,10 @@ def generate(state):
     """
     import torch
 
+    from modules.torch_utils import clear_torch_cache
+
+    clear_torch_cache()
+
     try:
         model_name = state['image_model_menu']