From 265c30a4cea73cda7c2b06a0f625ad3315743b11 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Thu, 10 Aug 2023 08:52:01 -0700
Subject: [PATCH] i

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

diff --git a/modules/core.py b/modules/core.py
index e3bda62e..08db86a4 100644
--- a/modules/core.py
+++ b/modules/core.py
@@ -72,6 +72,10 @@ def get_previewer(device, latent_format):
     return taesd
 
 
+def close_all_preview():
+    cv2.destroyAllWindows()
+
+
 @torch.no_grad()
 def ksampler(model, positive, negative, latent, seed=None, steps=30, cfg=9.0, sampler_name='euler_ancestral', scheduler='normal', denoise=1.0, disable_noise=False, start_step=None, last_step=None, force_full_denoise=False):
     seed = seed if isinstance(seed, int) else random.randint(1, 2 ** 64)
@@ -104,9 +108,6 @@ def ksampler(model, positive, negative, latent, seed=None, steps=30, cfg=9.0, sa
     out = latent.copy()
     out["samples"] = samples
 
-    if previewer:
-        cv2.destroyAllWindows()
-
     return out
 
 
diff --git a/modules/default_pipeline.py b/modules/default_pipeline.py
index b9df722b..366bc8c3 100644
--- a/modules/default_pipeline.py
+++ b/modules/default_pipeline.py
@@ -42,4 +42,7 @@ def process(positive_prompt, negative_prompt, width=1024, height=1024, batch_siz
     decoded_latent = core.decode_vae(vae=xl_refiner.vae, latent_image=sampled_latent)
 
     images = core.image_to_numpy(decoded_latent)
+
+    core.close_all_preview()
+    
     return images