add sdxl

2023-11-14 14:19:56 +08:00
parent 78c8d8dbdb
commit 8fbc8059e1
7 changed files with 246 additions and 14 deletions
--- a/lama_cleaner/const.py
+++ b/lama_cleaner/const.py
@@ -13,6 +13,7 @@ MPS_SUPPORT_MODELS = [
    "paint_by_example",
    "controlnet",
    "kandinsky2.2",
+    "sdxl",
 ]

 DEFAULT_MODEL = "lama"
@@ -23,6 +24,7 @@ AVAILABLE_MODELS = [
    "mat",
    "fcf",
    "sd1.5",
+    "sdxl",
    "anything4",
    "realisticVision1.4",
    "cv2",