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']