backend add freeu

2023-11-14 22:04:16 +08:00
parent 2c9a53da8e
commit bb98c91c8c
4 changed files with 33 additions and 3 deletions
--- a/lama_cleaner/const.py
+++ b/lama_cleaner/const.py
@@ -33,8 +33,10 @@ AVAILABLE_MODELS = [
    "paint_by_example",
    "instruct_pix2pix",
    "kandinsky2.2",
+    "sdxl"
 ]
 SD15_MODELS = ["sd1.5", "anything4", "realisticVision1.4"]
+MODELS_SUPPORT_FREEU = SD15_MODELS + ['sd2', "sdxl"]

 AVAILABLE_DEVICES = ["cuda", "cpu", "mps"]
 DEFAULT_DEVICE = "cuda"