More informative progress printing

1 files changed, 2 insertions, 0 deletions

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)