MJaheen commited on
Commit
d98f54b
·
1 Parent(s): bce3fee

make LCM model to be the default model

Browse files
Files changed (1) hide show
  1. 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
- "Pepe Fine-tuned (LoRA)": {
 
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": False
60
  },
61
- "Pepe + LCM (FAST)": {
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": True
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",