controlnet support load local ckpt

2023-04-01 09:23:37 +08:00
parent 5fd253b07e
commit 65f12b490a
3 changed files with 141 additions and 7 deletions
--- a/lama_cleaner/const.py
+++ b/lama_cleaner/const.py
@@ -8,6 +8,7 @@ MPS_SUPPORT_MODELS = [
    "realisticVision1.4",
    "sd2",
    "paint_by_example",
+    "controlnet"
 ]

 DEFAULT_MODEL = "lama"