Skip to content

Commit

Permalink
feat: automatically describe image on uov image upload (lllyasviel#1938)
Browse files Browse the repository at this point in the history
* feat: automatically describe image on uov image upload if prompt is empty

* feat: add argument to disable automatic uov image description

* feat: rename argument, disable by default

this prevents computers with low hardware specifications from being unnecessary blocked
  • Loading branch information
mashb1t authored and csokun committed Jun 16, 2024
1 parent 4914177 commit 797f805
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 0 deletions.
3 changes: 3 additions & 0 deletions args_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,9 @@
args_parser.parser.add_argument("--disable-preset-download", action='store_true',
help="Disables downloading models for presets", default=False)

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',
help="Always download newer models ", default=False)

Expand Down
9 changes: 9 additions & 0 deletions webui.py
Original file line number Diff line number Diff line change
Expand Up @@ -717,6 +717,15 @@ def trigger_describe(mode, img):
desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image],
outputs=[prompt, style_selections], show_progress=True, queue=True)

if args_manager.args.enable_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)

def dump_default_english_config():
from modules.localization import dump_english_config
Expand Down

0 comments on commit 797f805

Please sign in to comment.