Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: use non-prefixed uov attributes in async_worker

Browse Source
This commit is contained in:
Manuel Schmid
2024-06-24 21:49:04 +02:00
parent a245b67531
commit 1d3c0b9bca
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/async_worker.py
+4 -4
View File
@@ -968,10 +968,10 @@ def worker():
if 'vary' in goals: if 'vary' in goals:
img, denoising_strength, initial_latent, width, height, current_progress = apply_vary( img, denoising_strength, initial_latent, width, height, current_progress = apply_vary(
async_task, async_task.default_enhance_uov_method, denoising_strength, img, switch, current_progress) async_task, async_task.enhance_uov_method, denoising_strength, img, switch, current_progress)
if 'upscale' in goals: if 'upscale' in goals:
direct_return, img, denoising_strength, initial_latent, tiled, width, height, current_progress = apply_upscale( direct_return, img, denoising_strength, initial_latent, tiled, width, height, current_progress = apply_upscale(
async_task, img, async_task.default_enhance_uov_method, switch, current_progress) async_task, img, async_task.enhance_uov_method, switch, current_progress)
if direct_return: if direct_return:
d = [('Upscale (Fast)', 'upscale_fast', '2x')] d = [('Upscale (Fast)', 'upscale_fast', '2x')]
if modules.config.default_black_out_nsfw or async_task.black_out_nsfw: if modules.config.default_black_out_nsfw or async_task.black_out_nsfw:
@@ -1031,7 +1031,7 @@ def worker():
current_progress = int(base_progress + (100 - preparation_steps) / float(all_steps) * (done_steps_upscaling + done_steps_inpainting)) current_progress = int(base_progress + (100 - preparation_steps) / float(all_steps) * (done_steps_upscaling + done_steps_inpainting))
goals_enhance = [] goals_enhance = []
img, skip_prompt_processing, steps = prepare_upscale( img, skip_prompt_processing, steps = prepare_upscale(
async_task, goals_enhance, img, async_task.default_enhance_uov_method, async_task.performance_selection, async_task, goals_enhance, img, async_task.enhance_uov_method, async_task.performance_selection,
enhance_steps, current_progress) enhance_steps, current_progress)
steps, _, _, _ = apply_overrides(async_task, steps, height, width) steps, _, _, _ = apply_overrides(async_task, steps, height, width)
exception_result = '' exception_result = ''
@@ -1049,7 +1049,7 @@ def worker():
print('User skipped') print('User skipped')
async_task.last_stop = False async_task.last_stop = False
# also skip all enhance steps for this image, but add the steps to the progress bar # also skip all enhance steps for this image, but add the steps to the progress bar
if async_task.default_enhance_uov_processing_order == flags.enhancement_uov_before: if async_task.enhance_uov_processing_order == flags.enhancement_uov_before:
done_steps_inpainting += len(async_task.enhance_ctrls) * enhance_steps done_steps_inpainting += len(async_task.enhance_ctrls) * enhance_steps
exception_result = 'continue' exception_result = 'continue'
else: else: