From 3dc69a5293b045cd560467baef388836c7b66dd6 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Sat, 18 Nov 2023 12:14:33 +0100 Subject: [PATCH] fix progressbar call for nsfw output --- modules/async_worker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/async_worker.py b/modules/async_worker.py index 280895543..c57a12ae4 100644 --- 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')