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

feat: set inpaint_respective_field manually

Browse Source
This commit is contained in:
Manuel Schmid
2024-06-09 01:36:32 +02:00
parent 85b9005fa7
commit 9affa32583
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/async_worker.py
+10 -7
View File
@@ -430,8 +430,8 @@ def worker():
return initial_latent, width, height return initial_latent, width, height
def apply_inpaint(async_task, initial_latent, inpaint_head_model_path, inpaint_image, def apply_inpaint(async_task, initial_latent, inpaint_head_model_path, inpaint_image,
inpaint_mask, inpaint_parameterized, denoising_strength, switch, current_progress, inpaint_mask, inpaint_parameterized, denoising_strength, inpaint_respective_field, switch,
skip_apply_outpaint=False): current_progress, skip_apply_outpaint=False):
if not skip_apply_outpaint: if not skip_apply_outpaint:
inpaint_image, inpaint_mask = apply_outpaint(async_task, inpaint_image, inpaint_mask) inpaint_image, inpaint_mask = apply_outpaint(async_task, inpaint_image, inpaint_mask)
@@ -439,7 +439,7 @@ def worker():
image=inpaint_image, image=inpaint_image,
mask=inpaint_mask, mask=inpaint_mask,
use_fill=denoising_strength > 0.99, use_fill=denoising_strength > 0.99,
k=async_task.inpaint_respective_field k=inpaint_respective_field
) )
if async_task.debugging_inpaint_preprocessor: if async_task.debugging_inpaint_preprocessor:
yield_result(async_task, inpaint_worker.current_task.visualize_mask_processing(), async_task.black_out_nsfw, yield_result(async_task, inpaint_worker.current_task.visualize_mask_processing(), async_task.black_out_nsfw,
@@ -961,8 +961,9 @@ def worker():
denoising_strength, initial_latent, width, height = apply_inpaint(async_task, initial_latent, denoising_strength, initial_latent, width, height = apply_inpaint(async_task, initial_latent,
inpaint_head_model_path, inpaint_image, inpaint_head_model_path, inpaint_image,
inpaint_mask, inpaint_parameterized, inpaint_mask, inpaint_parameterized,
async_task.inpaint_strength, switch, async_task.inpaint_strength,
11) async_task.inpaint_respective_field,
switch, 11)
except EarlyReturnException: except EarlyReturnException:
return return
@@ -1044,13 +1045,15 @@ def worker():
yield_result(async_task, merged_masks, async_task.black_out_nsfw, False, yield_result(async_task, merged_masks, async_task.black_out_nsfw, False,
do_not_show_finished_images=len(tasks) == 1 or async_task.disable_intermediate_results) do_not_show_finished_images=len(tasks) == 1 or async_task.disable_intermediate_results)
# TODO make configurable # TODO make configurable
denoising_strength_adetailer = 0.5 denoising_strength_adetailer = 0.3
inpaint_respective_field_adetailer = 0.0
inpaint_head_model_path_adetailer = None inpaint_head_model_path_adetailer = None
inpaint_parameterized_adetailer = False inpaint_parameterized_adetailer = False
goals_adetailer = ['inpaint'] goals_adetailer = ['inpaint']
denoising_strength_adetailer, initial_latent_adetailer, width_adetailer, height_adetailer = apply_inpaint( denoising_strength_adetailer, initial_latent_adetailer, width_adetailer, height_adetailer = apply_inpaint(
async_task, None, inpaint_head_model_path_adetailer, img, merged_masks, async_task, None, inpaint_head_model_path_adetailer, img, merged_masks,
inpaint_parameterized_adetailer, denoising_strength_adetailer, switch, current_progress, True) inpaint_parameterized_adetailer, denoising_strength_adetailer,
inpaint_respective_field_adetailer, switch, current_progress, True)
process_task(all_steps, async_task, callback, controlnet_canny_path, controlnet_cpds_path, process_task(all_steps, async_task, callback, controlnet_canny_path, controlnet_cpds_path,
current_task_id, denoising_strength_adetailer, final_scheduler_name, goals_adetailer, current_task_id, denoising_strength_adetailer, final_scheduler_name, goals_adetailer,