1580158
/
sd-webui-inpaint-anything
mirror of https://github.com/Uminosachi/sd-webui-inpaint-anything.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Move SDXL VAE validation step after model loading

main
Uminosachi 2024-06-09 17:06:57 +09:00
parent 67b7f80f69
commit 4e5c2c7a91
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
scripts/inpaint_anything.py
View File

@ -746,6 +746,13 @@ def run_webui_inpaint(input_image, sel_mask,
ia_logging.error("The sizes of the image and mask do not match")
return
info = get_closet_checkpoint_match(webui_model_id)
if info is None:
ia_logging.error(f"No model found: {webui_model_id}")
return
await_backup_reload_ckpt_info(info=info)
if not getattr(shared.sd_model, "is_sdxl", False) and "sdxl_vae" in getattr(shared.opts, "sd_vae", ""):
ia_logging.error("The SDXL VAE is not compatible with the inpainting model")
ret_image = draw_text_image(
@ -757,13 +764,6 @@ def run_webui_inpaint(input_image, sel_mask,
save_mask_image(mask_image, webui_save_mask_chk)
info = get_closet_checkpoint_match(webui_model_id)
if info is None:
ia_logging.error(f"No model found: {webui_model_id}")
return
await_backup_reload_ckpt_info(info=info)
init_image, mask_image = auto_resize_to_pil(input_image, mask_image)
width, height = init_image.size