From 9992b88c0e719362bdcbe61dd610af9c515fc641 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Thu, 10 Aug 2023 21:07:09 -0700
Subject: [PATCH] i

---
 modules/default_pipeline.py | 3 ---
 webui.py                    | 2 ++
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/modules/default_pipeline.py b/modules/default_pipeline.py
index 0bd951a9..8e4cd0b0 100644
--- a/modules/default_pipeline.py
+++ b/modules/default_pipeline.py
@@ -3,7 +3,6 @@ import os
 import torch
 
 from modules.path import modelfile_path, lorafile_path
-from modules.cv2win32 import close_all_preview
 
 
 xl_base_filename = os.path.join(modelfile_path, 'sd_xl_base_1.0.safetensors')
@@ -44,6 +43,4 @@ def process(positive_prompt, negative_prompt, steps, switch, width, height, imag
 
     images = core.image_to_numpy(decoded_latent)
 
-    close_all_preview()
-
     return images
diff --git a/webui.py b/webui.py
index 20223c2e..7ac508e1 100644
--- a/webui.py
+++ b/webui.py
@@ -3,6 +3,7 @@ import random
 
 from modules.sdxl_styles import apply_style, style_keys, aspect_ratios
 from modules.default_pipeline import process
+from modules.cv2win32 import close_all_preview
 
 
 def generate_clicked(prompt, negative_prompt, style_selction, performance_selction,
@@ -29,6 +30,7 @@ def generate_clicked(prompt, negative_prompt, style_selction, performance_selcti
         seed += 1
         results += imgs
 
+    close_all_preview()
     return results