From 51dc53b5c810b86ce46fb79977f7a8c997b5a93d Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Thu, 20 Jun 2024 02:49:22 +0200 Subject: [PATCH] feat: save and output upscaled image with method fast in enhance --- modules/async_worker.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/modules/async_worker.py b/modules/async_worker.py index 55081cb4..6558aa95 100644 --- a/modules/async_worker.py +++ b/modules/async_worker.py @@ -968,6 +968,14 @@ def worker(): direct_return, img, denoising_strength, initial_latent, tiled, width, height, current_progress = apply_upscale( async_task, img, async_task.enhance_uov_method, switch, current_progress) if direct_return: + d = [('Upscale (Fast)', 'upscale_fast', '2x')] + if modules.config.default_black_out_nsfw or async_task.black_out_nsfw: + progressbar(async_task, current_progress, 'Checking for NSFW content ...') + async_task.uov_input_image = default_censor(async_task.uov_input_image) + progressbar(async_task, current_progress, f'Saving image {current_task_id + 1}/{total_count} to system ...') + uov_input_image_path = log(async_task.uov_input_image, d, output_format=async_task.output_format) + yield_result(async_task, uov_input_image_path, current_progress, async_task.black_out_nsfw, False, + do_not_show_finished_images=True) return current_progress, img if 'inpaint' in goals and inpaint_parameterized: