mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
minor fixes from audit
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -230,6 +230,8 @@ def img2img(id_task: str, state: str, mode: int,
|
||||
elif mode == 4: # inpaint upload mask
|
||||
if init_img_inpaint is None:
|
||||
return [], '', '', 'Error: inpaint image not provided'
|
||||
if init_mask_inpaint is None:
|
||||
return [], '', '', 'Error: inpaint mask not provided'
|
||||
image = init_img_inpaint
|
||||
mask = init_mask_inpaint
|
||||
elif mode == 5: # process batch
|
||||
|
||||
@@ -445,7 +445,7 @@ def create_ui():
|
||||
# Load/Unload model buttons
|
||||
vlm_load_btn.click(fn=vqa.load_model, inputs=[vlm_model], outputs=[])
|
||||
vlm_unload_btn.click(fn=vqa.unload_model, inputs=[], outputs=[])
|
||||
analyze_load_btn.click(fn=vqa.load_model, inputs=[vlm_model], outputs=[])
|
||||
analyze_load_btn.click(fn=vqa.load_model, inputs=[analyze_model], outputs=[])
|
||||
analyze_unload_btn.click(fn=vqa.unload_model, inputs=[], outputs=[])
|
||||
|
||||
def tagger_load_wrapper(model_name):
|
||||
|
||||
Reference in New Issue
Block a user