Spaces:
Sleeping
Sleeping
make LCM model to be the default model
Browse files- src/model/config.py +6 -5
src/model/config.py
CHANGED
|
@@ -51,20 +51,21 @@ class ModelConfig:
|
|
| 51 |
if self.AVAILABLE_MODELS is None:
|
| 52 |
self.AVAILABLE_MODELS = {
|
| 53 |
# Primary fine-tuned model - Best quality, trained on Pepe dataset
|
| 54 |
-
|
|
|
|
| 55 |
"base": "runwayml/stable-diffusion-v1-5",
|
| 56 |
"lora": "MJaheen/Pepe_The_Frog_model_v1_lora",
|
|
|
|
| 57 |
"trigger_word": "pepe_style_frog",
|
| 58 |
"use_lora": True,
|
| 59 |
-
"use_lcm":
|
| 60 |
},
|
| 61 |
-
"Pepe
|
| 62 |
"base": "runwayml/stable-diffusion-v1-5",
|
| 63 |
"lora": "MJaheen/Pepe_The_Frog_model_v1_lora",
|
| 64 |
-
"lcm_lora": "latent-consistency/lcm-lora-sdv1-5",
|
| 65 |
"trigger_word": "pepe_style_frog",
|
| 66 |
"use_lora": True,
|
| 67 |
-
"use_lcm":
|
| 68 |
},
|
| 69 |
"Base SD 1.5": {
|
| 70 |
"base": "runwayml/stable-diffusion-v1-5",
|
|
|
|
| 51 |
if self.AVAILABLE_MODELS is None:
|
| 52 |
self.AVAILABLE_MODELS = {
|
| 53 |
# Primary fine-tuned model - Best quality, trained on Pepe dataset
|
| 54 |
+
|
| 55 |
+
"Pepe + LCM (FAST)": {
|
| 56 |
"base": "runwayml/stable-diffusion-v1-5",
|
| 57 |
"lora": "MJaheen/Pepe_The_Frog_model_v1_lora",
|
| 58 |
+
"lcm_lora": "latent-consistency/lcm-lora-sdv1-5",
|
| 59 |
"trigger_word": "pepe_style_frog",
|
| 60 |
"use_lora": True,
|
| 61 |
+
"use_lcm": True
|
| 62 |
},
|
| 63 |
+
"Pepe Fine-tuned (LoRA)": {
|
| 64 |
"base": "runwayml/stable-diffusion-v1-5",
|
| 65 |
"lora": "MJaheen/Pepe_The_Frog_model_v1_lora",
|
|
|
|
| 66 |
"trigger_word": "pepe_style_frog",
|
| 67 |
"use_lora": True,
|
| 68 |
+
"use_lcm": False
|
| 69 |
},
|
| 70 |
"Base SD 1.5": {
|
| 71 |
"base": "runwayml/stable-diffusion-v1-5",
|