Merge pull request #6648 from vladmandic/progress-description

Set TQDM progress bar and state textinfo description

1 files changed, 6 insertions, 1 deletions

diff --git a/modules/textual_inversion/preprocess.py b/modules/textual_inversion/preprocess.py
index feb876c6..3c1042ad 100644
--- a/modules/textual_inversion/preprocess.py
+++ b/modules/textual_inversion/preprocess.py
@@ -135,7 +135,8 @@ def preprocess_work(process_src, process_dst, process_width, process_height, pre
     params.process_caption_deepbooru = process_caption_deepbooru

     params.preprocess_txt_action = preprocess_txt_action

 

-    for index, imagefile in enumerate(tqdm.tqdm(files)):

+    pbar = tqdm.tqdm(files)

+    for index, imagefile in enumerate(pbar):

         params.subindex = 0

         filename = os.path.join(src, imagefile)

         try:

@@ -143,6 +144,10 @@ def preprocess_work(process_src, process_dst, process_width, process_height, pre
         except Exception:

             continue

 

+        description = f"Preprocessing [Image {index}/{len(files)}]"

+        pbar.set_description(description)

+        shared.state.textinfo = description

+

         params.src = filename

 

         existing_caption = None