happyz
/
Fooocus
mirror of https://github.com/lllyasviel/Fooocus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: save and output upscaled image with method fast in enhance

This commit is contained in:
Manuel Schmid 2024-06-20 02:49:22 +02:00
parent eada51dee4
commit 51dc53b5c8
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/async_worker.py
View File

@ -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: