feat: add argument to disable automatic uov image description

This commit is contained in:
Manuel Schmid
2024-05-17 17:42:15 +02:00
parent 72f59c9554
commit cb34100295
2 changed files with 11 additions and 7 deletions
+3
View File
@@ -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', 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("--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', 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)
+1
View File
@@ -705,6 +705,7 @@ 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 not args_manager.args.disable_describe_uov_image:
def trigger_uov_describe(mode, img, prompt): def trigger_uov_describe(mode, img, prompt):
# keep prompt if not empty # keep prompt if not empty
if prompt == '': if prompt == '':