From f211c498b9d496222aa36360193e0f690053dfa4 Mon Sep 17 00:00:00 2001
From: JohannesGaessler <johannesg@5d6.de>
Date: Thu, 8 Sep 2022 15:37:13 +0200
Subject: More informative progress printing

---
 modules/img2img.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'modules/img2img.py')

diff --git a/modules/img2img.py b/modules/img2img.py
index c2392305..b5f3def3 100644
--- a/modules/img2img.py
+++ b/modules/img2img.py
@@ -52,6 +52,7 @@ def img2img(prompt: str, init_img, init_img_with_mask, steps: int, sampler_index
         inpainting_mask_invert=inpainting_mask_invert,
         extra_generation_params={"Denoising Strength": denoising_strength}
     )
+    print(f"\nimg2img: {prompt}", file=shared.progress_print_out)
 
     if is_loopback:
         output_images, info = None, None
@@ -168,5 +169,6 @@ def img2img(prompt: str, init_img, init_img_with_mask, steps: int, sampler_index
         if processed is None:
             processed = process_images(p)
 
+    shared.total_tqdm.clear()
 
     return processed.images, processed.js(), plaintext_to_html(processed.info)
-- 
cgit v1.2.3