From 14dbc3488e5780be50a7b6211742ec2137cdc503 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Thu, 4 Dec 2025 05:32:58 -0800 Subject: [PATCH] Image: Clear the torch cache after generation, not before --- modules/ui_image_generation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py index fdf1af86..7c393f13 100644 --- a/modules/ui_image_generation.py +++ b/modules/ui_image_generation.py @@ -670,8 +670,6 @@ def generate(state): from modules.torch_utils import clear_torch_cache, get_device - clear_torch_cache() - try: model_name = state['image_model_menu'] @@ -772,11 +770,13 @@ def generate(state): logger.info(f'Generated {total_images} {"image" if total_images == 1 else "images"} in {(t1 - t0):.2f} seconds ({total_steps / (t1 - t0):.2f} steps/s, seed {seed})') yield all_images + clear_torch_cache() except Exception as e: logger.error(f"Image generation failed: {e}") traceback.print_exc() yield [] + clear_torch_cache() def load_image_model_wrapper(model_name, dtype, attn_backend, cpu_offload, compile_model, quant_method):