minor fixes from audit

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-07-01 09:52:49 +02:00
parent 7e21e17d08
commit 02bb44aa90
7 changed files with 34 additions and 14 deletions
+2
View File
@@ -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
+1 -1
View File
@@ -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):