update

2023-02-06 22:00:47 +08:00
parent 24bff09534
commit 3f6bc8fada
9 changed files with 307 additions and 91 deletions
--- a/lama_cleaner/const.py
+++ b/lama_cleaner/const.py
@@ -1,5 +1,12 @@
 import os

+MPS_SUPPORT_MODELS = [
+    "instruct_pix2pix",
+    "sd1.5",
+    "sd2",
+    "paint_by_example"
+]
+
 DEFAULT_MODEL = "lama"
 AVAILABLE_MODELS = [
    "lama",