mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
feat: add config for enhance upv checkbox and method, prefix with default_
This commit is contained in:
@@ -968,10 +968,10 @@ def worker():
|
||||
|
||||
if 'vary' in goals:
|
||||
img, denoising_strength, initial_latent, width, height, current_progress = apply_vary(
|
||||
async_task, async_task.enhance_uov_method, denoising_strength, img, switch, current_progress)
|
||||
async_task, async_task.default_enhance_uov_method, denoising_strength, img, switch, current_progress)
|
||||
if 'upscale' in goals:
|
||||
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)
|
||||
async_task, img, async_task.default_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:
|
||||
@@ -1031,7 +1031,7 @@ def worker():
|
||||
current_progress = int(base_progress + (100 - preparation_steps) / float(all_steps) * (done_steps_upscaling + done_steps_inpainting))
|
||||
goals_enhance = []
|
||||
img, skip_prompt_processing, steps = prepare_upscale(
|
||||
async_task, goals_enhance, img, async_task.enhance_uov_method, async_task.performance_selection,
|
||||
async_task, goals_enhance, img, async_task.default_enhance_uov_method, async_task.performance_selection,
|
||||
enhance_steps, current_progress)
|
||||
steps, _, _, _ = apply_overrides(async_task, steps, height, width)
|
||||
exception_result = ''
|
||||
@@ -1049,7 +1049,7 @@ def worker():
|
||||
print('User skipped')
|
||||
async_task.last_stop = False
|
||||
# also skip all enhance steps for this image, but add the steps to the progress bar
|
||||
if async_task.enhance_uov_processing_order == flags.enhancement_uov_before:
|
||||
if async_task.default_enhance_uov_processing_order == flags.enhancement_uov_before:
|
||||
done_steps_inpainting += len(async_task.enhance_ctrls) * enhance_steps
|
||||
exception_result = 'continue'
|
||||
else:
|
||||
|
||||
+14
-2
@@ -515,13 +515,25 @@ default_enhance_tabs = get_config_item_or_set_default(
|
||||
validator=lambda x: isinstance(x, int) and 1 <= x <= 5,
|
||||
expected_type=int
|
||||
)
|
||||
enhance_uov_processing_order = get_config_item_or_set_default(
|
||||
default_enhance_checkbox = get_config_item_or_set_default(
|
||||
key='default_enhance_checkbox',
|
||||
default_value=False,
|
||||
validator=lambda x: isinstance(x, bool),
|
||||
expected_type=bool
|
||||
)
|
||||
default_enhance_uov_method = get_config_item_or_set_default(
|
||||
key='default_enhance_uov_method',
|
||||
default_value=modules.flags.disabled,
|
||||
validator=lambda x: x in modules.flags.uov_list,
|
||||
expected_type=int
|
||||
)
|
||||
default_enhance_uov_processing_order = get_config_item_or_set_default(
|
||||
key='default_enhance_uov_processing_order',
|
||||
default_value=modules.flags.enhancement_uov_before,
|
||||
validator=lambda x: x in modules.flags.enhancement_uov_processing_order,
|
||||
expected_type=int
|
||||
)
|
||||
enhance_uov_prompt_type = get_config_item_or_set_default(
|
||||
default_enhance_uov_prompt_type = get_config_item_or_set_default(
|
||||
key='default_enhance_uov_prompt_type',
|
||||
default_value=modules.flags.enhancement_uov_prompt_type_original,
|
||||
validator=lambda x: x in modules.flags.enhancement_uov_prompt_types,
|
||||
|
||||
+1
-3
@@ -8,9 +8,7 @@ upscale_15 = 'Upscale (1.5x)'
|
||||
upscale_2 = 'Upscale (2x)'
|
||||
upscale_fast = 'Upscale (Fast 2x)'
|
||||
|
||||
uov_list = [
|
||||
disabled, subtle_variation, strong_variation, upscale_15, upscale_2, upscale_fast
|
||||
]
|
||||
uov_list = [disabled, subtle_variation, strong_variation, upscale_15, upscale_2, upscale_fast]
|
||||
|
||||
enhancement_uov_before = "Before First Enhancement"
|
||||
enhancement_uov_after = "After Last Enhancement"
|
||||
|
||||
Reference in New Issue
Block a user