add load local model

2023-03-29 22:05:34 +08:00
parent 61df5f69b3
commit f2e90d3f84
7 changed files with 131 additions and 10 deletions
--- a/lama_cleaner/const.py
+++ b/lama_cleaner/const.py
@@ -52,6 +52,10 @@ SD_CONTROLNET_HELP = """
 Run Stable Diffusion 1.5 inpainting model with Canny ControlNet control.
 """

+SD_LOCAL_MODEL_HELP = """
+Load Stable Diffusion 1.5 model(ckpt/safetensors) from local path.
+"""
+
 LOCAL_FILES_ONLY_HELP = """
 Use local files only, not connect to Hugging Face server. (sd/paint_by_example)
 """