add lcm lora

2023-11-15 08:50:35 +08:00
parent bb98c91c8c
commit 58b931fdb2
6 changed files with 33 additions and 12 deletions
--- a/lama_cleaner/const.py
+++ b/lama_cleaner/const.py
@@ -37,6 +37,7 @@ AVAILABLE_MODELS = [
 ]
 SD15_MODELS = ["sd1.5", "anything4", "realisticVision1.4"]
 MODELS_SUPPORT_FREEU = SD15_MODELS + ['sd2', "sdxl"]
+MODELS_SUPPORT_LCM_LORA = SD15_MODELS + ["sdxl"]

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