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

fix: use 0 steps when upscaling with skip_prompt_processing

This commit is contained in:
Manuel Schmid 2024-06-22 01:45:53 +02:00
parent 7ed98521ec
commit bd3a29ba68
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
modules/async_worker.py
View File

@ -930,6 +930,7 @@ def worker():
goals.append('upscale')
if 'fast' in uov_method:
skip_prompt_processing = True
steps = 0
else:
steps = performance.steps_uov()
@ -977,7 +978,7 @@ def worker():
progressbar(async_task, current_progress, f'Saving image {current_task_id + 1}/{total_count} to system ...')
uov_image_path = log(img, d, output_format=async_task.output_format)
yield_result(async_task, uov_image_path, current_progress, async_task.black_out_nsfw, False,
do_not_show_finished_images=True)
do_not_show_finished_images=not show_intermediate_results or async_task.disable_intermediate_results)
return current_progress, img
if 'inpaint' in goals and inpaint_parameterized:
@ -1122,7 +1123,7 @@ def worker():
if 'vary' in goals:
async_task.uov_input_image, denoising_strength, initial_latent, width, height, current_progress = apply_vary(
async_task, async_task.uov_method, denoising_strength, switch, async_task.uov_input_image,
async_task, async_task.uov_method, denoising_strength, async_task.uov_input_image, switch,
current_progress)
if 'upscale' in goals:
@ -1178,7 +1179,13 @@ def worker():
all_steps = steps * async_task.image_number
if async_task.enhance_checkbox and async_task.enhance_uov_method != flags.disabled.casefold():
enhance_upscale_steps, _, _, _ = apply_overrides(async_task, async_task.performance_selection.steps_uov(), height, width)
enhance_upscale_steps = async_task.performance_selection.steps()
if 'upscale' in async_task.enhance_uov_method:
if 'fast' in async_task.enhance_uov_method:
enhance_upscale_steps = 0
else:
enhance_upscale_steps = async_task.performance_selection.steps_uov()
enhance_upscale_steps, _, _, _ = apply_overrides(async_task, enhance_upscale_steps, height, width)
enhance_upscale_steps_total = async_task.image_number * enhance_upscale_steps
all_steps += enhance_upscale_steps_total