From cb34100295e8529b89a83d656a3dfbbaf169c9b7 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Fri, 17 May 2024 17:42:15 +0200 Subject: [PATCH 1/2] feat: add argument to disable automatic uov image description --- args_manager.py | 3 +++ webui.py | 15 ++++++++------- 2 files changed, 11 insertions(+), 7 deletions(-) diff --git a/args_manager.py b/args_manager.py index 6a3ae9dc..336df08c 100644 --- a/args_manager.py +++ b/args_manager.py @@ -31,6 +31,9 @@ args_parser.parser.add_argument("--disable-metadata", action='store_true', args_parser.parser.add_argument("--disable-preset-download", action='store_true', help="Disables downloading models for presets", default=False) +args_parser.parser.add_argument("--disable-describe-uov-image", action='store_true', + help="Disables automatic description of uov images when prompt is empty", default=False) + args_parser.parser.add_argument("--always-download-new-model", action='store_true', help="Always download newer models ", default=False) diff --git a/webui.py b/webui.py index cf163f7c..ff4d7089 100644 --- a/webui.py +++ b/webui.py @@ -705,14 +705,15 @@ with shared.gradio_root: desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image], outputs=[prompt, style_selections], show_progress=True, queue=True) - def trigger_uov_describe(mode, img, prompt): - # keep prompt if not empty - if prompt == '': - return trigger_describe(mode, img) - return gr.update(), gr.update() + if not args_manager.args.disable_describe_uov_image: + def trigger_uov_describe(mode, img, prompt): + # keep prompt if not empty + if prompt == '': + return trigger_describe(mode, img) + return gr.update(), gr.update() - uov_input_image.upload(trigger_uov_describe, inputs=[desc_method, uov_input_image, prompt], - outputs=[prompt, style_selections], show_progress=True, queue=True) + uov_input_image.upload(trigger_uov_describe, inputs=[desc_method, uov_input_image, prompt], + outputs=[prompt, style_selections], show_progress=True, queue=True) def dump_default_english_config(): from modules.localization import dump_english_config From 24721a85112f10242da5872f76b54d4a6308d057 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Fri, 17 May 2024 18:19:48 +0200 Subject: [PATCH 2/2] feat: rename argument, disable by default this prevents computers with low hardware specifications from being unnecessary blocked --- args_manager.py | 2 +- webui.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/args_manager.py b/args_manager.py index 336df08c..e023da27 100644 --- a/args_manager.py +++ b/args_manager.py @@ -31,7 +31,7 @@ args_parser.parser.add_argument("--disable-metadata", action='store_true', args_parser.parser.add_argument("--disable-preset-download", action='store_true', help="Disables downloading models for presets", default=False) -args_parser.parser.add_argument("--disable-describe-uov-image", action='store_true', +args_parser.parser.add_argument("--enable-describe-uov-image", action='store_true', help="Disables automatic description of uov images when prompt is empty", default=False) args_parser.parser.add_argument("--always-download-new-model", action='store_true', diff --git a/webui.py b/webui.py index ff4d7089..46538c94 100644 --- a/webui.py +++ b/webui.py @@ -705,7 +705,7 @@ with shared.gradio_root: desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image], outputs=[prompt, style_selections], show_progress=True, queue=True) - if not args_manager.args.disable_describe_uov_image: + if args_manager.args.enable_describe_uov_image: def trigger_uov_describe(mode, img, prompt): # keep prompt if not empty if prompt == '':