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: rename arg --enable-describe-uov-image to --enable-auto-describe-image, add handling for enhance input image

Browse Source
This commit is contained in:
Manuel Schmid
2024-06-22 00:07:32 +02:00
parent ef236339f0
commit ec533f0c42
2 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
args_manager.py
+2 -2
View File
@@ -28,8 +28,8 @@ args_parser.parser.add_argument("--disable-metadata", action='store_true',
args_parser.parser.add_argument("--disable-preset-download", action='store_true', args_parser.parser.add_argument("--disable-preset-download", action='store_true',
help="Disables downloading models for presets", default=False) help="Disables downloading models for presets", default=False)
args_parser.parser.add_argument("--enable-describe-uov-image", action='store_true', args_parser.parser.add_argument("--enable-auto-describe-image", action='store_true',
help="Disables automatic description of uov images when prompt is empty", default=False) help="Enables automatic description of uov and enhance image when prompt is empty", default=False)
args_parser.parser.add_argument("--always-download-new-model", action='store_true', args_parser.parser.add_argument("--always-download-new-model", action='store_true',
help="Always download newer models ", default=False) help="Always download newer models ", default=False)
webui.py
+7 -4
View File
@@ -1001,15 +1001,18 @@ with shared.gradio_root:
desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image], desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image],
outputs=[prompt, style_selections], show_progress=True, queue=True) outputs=[prompt, style_selections], show_progress=True, queue=True)
if args_manager.args.enable_describe_uov_image: if args_manager.args.enable_auto_describe_image:
def trigger_uov_describe(mode, img, prompt): def trigger_auto_describe(mode, img, prompt):
# keep prompt if not empty # keep prompt if not empty
if prompt == '': if prompt == '':
return trigger_describe(mode, img) return trigger_describe(mode, img)
return gr.update(), gr.update() return gr.update(), gr.update()
uov_input_image.upload(trigger_uov_describe, inputs=[desc_method, uov_input_image, prompt], uov_input_image.upload(trigger_auto_describe, inputs=[desc_method, uov_input_image, prompt],
outputs=[prompt, style_selections], show_progress=True, queue=True) outputs=[prompt, style_selections], show_progress=True, queue=True)
enhance_input_image.upload(trigger_auto_describe, inputs=[desc_method, enhance_input_image, prompt],
outputs=[prompt, style_selections], show_progress=True, queue=True)
def dump_default_english_config(): def dump_default_english_config():
from modules.localization import dump_english_config from modules.localization import dump_english_config