diff --git a/README.md b/README.md index 9599405..04b3c32 100644 --- a/README.md +++ b/README.md @@ -66,7 +66,7 @@ To download the model: ### Inpainting webui Tab * This tab becomes accessible when you have an inpainting model. -* The model required should have `inpaint` in its filename and must be located in the `stable-diffusion-webui/models` directory. +* The model required should have `inpaint` (case-insensitive) in its filename and must be located in the `stable-diffusion-webui/models` directory. * Once the model is recognized, it can be selected from the Inpainting Model ID dropdown list. ### ControlNet Inpaint Tab diff --git a/scripts/inpaint_anything.py b/scripts/inpaint_anything.py index a0462eb..2ef0a03 100644 --- a/scripts/inpaint_anything.py +++ b/scripts/inpaint_anything.py @@ -19,7 +19,6 @@ import cv2 from huggingface_hub import snapshot_download from lama_cleaner.model_manager import ModelManager from lama_cleaner.schema import Config, HDStrategy, LDMSampler, SDSampler -# print("platform:", platform.system()) import modules.scripts as scripts from modules import shared, script_callbacks @@ -908,8 +907,7 @@ def on_ui_tabs(): webui_inpaint_enabled = False list_ckpt = shared.list_checkpoint_tiles() - webui_model_ids = [ckpt for ckpt in list_ckpt if "inpaint" in ckpt] - print(webui_model_ids) + webui_model_ids = [ckpt for ckpt in list_ckpt if "inpaint" in ckpt.lower()] if len(webui_model_ids) > 0: webui_inpaint_enabled = True