fix progressbar call for nsfw output

lllyasviel · Nov 18, 2023 · 3dc69a5 · 3dc69a5
1 parent 6893c8d
commit 3dc69a5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/modules/async_worker.py b/modules/async_worker.py
@@ -56,12 +56,12 @@ def progressbar(async_task, number, text):
         print(f'[Fooocus] {text}')
         async_task.yields.append(['preview', (number, text, None)])
 
-    def yield_result(async_task, imgs, do_not_show_finished_images=False):
+    def yield_result(async_task, imgs, do_not_show_finished_images=False, progressbar_index=13):
         if not isinstance(imgs, list):
             imgs = [imgs]
 
         if modules.config.default_black_out_nsfw:
-            progressbar(progressbar_index, 'Checking for NSFW content ...')
+            progressbar(async_task, progressbar_index, 'Checking for NSFW content ...')
             imgs = censor_batch(imgs)
 
         async_task.results = async_task.results + imgs
@@ -725,7 +725,7 @@ def callback(step, x0, x, total_steps, y):
                             d.append((f'LoRA [{n}] weight', w))
                     log(x, d, single_line_number=3)
 
-                yield_result(async_task, imgs, do_not_show_finished_images=len(tasks) == 1, progressbar_index=int(15.0 + 85.0 * float((current_task_id + 1) * steps) / float(all_steps))))
+                yield_result(async_task, imgs, do_not_show_finished_images=len(tasks) == 1, progressbar_index=int(15.0 + 85.0 * float((current_task_id + 1) * steps) / float(all_steps)))
             except fcbh.model_management.InterruptProcessingException as e:
                 if shared.last_stop == 'skip':
                     print('User skipped')