From f26d2e76ec878dea7c1e8474cb58bd0d310299b0 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 | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/async_worker.py b/modules/async_worker.py index 280895543..ac175e81f 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