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: extract attribute generate_image_grid

Browse Source
This commit is contained in:
Manuel Schmid
2024-01-22 18:23:10 +01:00
parent d72573aca4
commit 217be190bb
3 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/advanced_parameters.py
+4 -4
View File
@@ -1,15 +1,15 @@
generate_image_grid, overwrite_step, overwrite_switch, overwrite_width, overwrite_height, \ overwrite_step, overwrite_switch, overwrite_width, overwrite_height, \
overwrite_vary_strength, overwrite_upscale_strength, \ overwrite_vary_strength, overwrite_upscale_strength, \
mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint, \ mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint, \
debugging_cn_preprocessor, skipping_cn_preprocessor, controlnet_softness, canny_low_threshold, canny_high_threshold, \ debugging_cn_preprocessor, skipping_cn_preprocessor, controlnet_softness, canny_low_threshold, canny_high_threshold, \
refiner_swap_method, \ refiner_swap_method, \
freeu_enabled, freeu_b1, freeu_b2, freeu_s1, freeu_s2, \ freeu_enabled, freeu_b1, freeu_b2, freeu_s1, freeu_s2, \
debugging_inpaint_preprocessor, inpaint_disable_initial_latent, inpaint_engine, inpaint_strength, inpaint_respective_field, \ debugging_inpaint_preprocessor, inpaint_disable_initial_latent, inpaint_engine, inpaint_strength, inpaint_respective_field, \
inpaint_mask_upload_checkbox, invert_mask_checkbox, inpaint_erode_or_dilate = [None] * 28 inpaint_mask_upload_checkbox, invert_mask_checkbox, inpaint_erode_or_dilate = [None] * 27
def set_all_advanced_parameters(*args): def set_all_advanced_parameters(*args):
global generate_image_grid, overwrite_step, overwrite_switch, overwrite_width, overwrite_height, \ global overwrite_step, overwrite_switch, overwrite_width, overwrite_height, \
overwrite_vary_strength, overwrite_upscale_strength, \ overwrite_vary_strength, overwrite_upscale_strength, \
mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint, \ mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint, \
debugging_cn_preprocessor, skipping_cn_preprocessor, controlnet_softness, canny_low_threshold, canny_high_threshold, \ debugging_cn_preprocessor, skipping_cn_preprocessor, controlnet_softness, canny_low_threshold, canny_high_threshold, \
@@ -18,7 +18,7 @@ def set_all_advanced_parameters(*args):
debugging_inpaint_preprocessor, inpaint_disable_initial_latent, inpaint_engine, inpaint_strength, inpaint_respective_field, \ debugging_inpaint_preprocessor, inpaint_disable_initial_latent, inpaint_engine, inpaint_strength, inpaint_respective_field, \
inpaint_mask_upload_checkbox, invert_mask_checkbox, inpaint_erode_or_dilate inpaint_mask_upload_checkbox, invert_mask_checkbox, inpaint_erode_or_dilate
generate_image_grid, overwrite_step, overwrite_switch, overwrite_width, overwrite_height, \ overwrite_step, overwrite_switch, overwrite_width, overwrite_height, \
overwrite_vary_strength, overwrite_upscale_strength, \ overwrite_vary_strength, overwrite_upscale_strength, \
mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint, \ mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint, \
debugging_cn_preprocessor, skipping_cn_preprocessor, controlnet_softness, canny_low_threshold, canny_high_threshold, \ debugging_cn_preprocessor, skipping_cn_preprocessor, controlnet_softness, canny_low_threshold, canny_high_threshold, \
modules/async_worker.py
+4 -4
View File
@@ -69,9 +69,6 @@ def worker():
return return
def build_image_wall(async_task): def build_image_wall(async_task):
if not advanced_parameters.generate_image_grid:
return
results = async_task.results results = async_task.results
if len(results) < 2: if len(results) < 2:
@@ -824,9 +821,12 @@ def worker():
time.sleep(0.01) time.sleep(0.01)
if len(async_tasks) > 0: if len(async_tasks) > 0:
task = async_tasks.pop(0) task = async_tasks.pop(0)
generate_image_grid = task.args.pop(0)
try: try:
handler(task) handler(task)
build_image_wall(task) if generate_image_grid:
build_image_wall(task)
task.yields.append(['finish', task.results]) task.yields.append(['finish', task.results])
pipeline.prepare_text_encoder(async_call=True) pipeline.prepare_text_encoder(async_call=True)
except: except:
webui.py
+3 -2
View File
@@ -446,7 +446,7 @@ with shared.gradio_root:
freeu_s2 = gr.Slider(label='S2', minimum=0, maximum=4, step=0.01, value=0.95) freeu_s2 = gr.Slider(label='S2', minimum=0, maximum=4, step=0.01, value=0.95)
freeu_ctrls = [freeu_enabled, freeu_b1, freeu_b2, freeu_s1, freeu_s2] freeu_ctrls = [freeu_enabled, freeu_b1, freeu_b2, freeu_s1, freeu_s2]
adps = [generate_image_grid, overwrite_step, overwrite_switch, overwrite_width, overwrite_height, adps = [overwrite_step, overwrite_switch, overwrite_width, overwrite_height,
overwrite_vary_strength, overwrite_upscale_strength, overwrite_vary_strength, overwrite_upscale_strength,
mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint, mixing_image_prompt_and_vary_upscale, mixing_image_prompt_and_inpaint,
debugging_cn_preprocessor, skipping_cn_preprocessor, controlnet_softness, debugging_cn_preprocessor, skipping_cn_preprocessor, controlnet_softness,
@@ -518,7 +518,8 @@ with shared.gradio_root:
inpaint_strength, inpaint_respective_field inpaint_strength, inpaint_respective_field
], show_progress=False, queue=False) ], show_progress=False, queue=False)
ctrls = [ ctrls = [generate_image_grid]
ctrls += [
prompt, negative_prompt, style_selections, prompt, negative_prompt, style_selections,
performance_selection, aspect_ratios_selection, image_number, image_seed, sharpness, guidance_scale performance_selection, aspect_ratios_selection, image_number, image_seed, sharpness, guidance_scale
] ]