Make inpainting model search case-insensitive

pull/79/head v1.9.0
Uminosachi 2023-06-19 12:28:05 +09:00
parent 3c8910e732
commit 084e5f85bd
2 changed files with 2 additions and 4 deletions

View File

@ -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

View File

@ -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