Hugging Face's logo

Spaces:
multimodalart
/
civitai-to-hf
Running

App Files Community
6
multimodalart HF staff commited on
Commit
ff31e08
•
1 Parent(s): 93009ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -80,7 +80,9 @@ def extract_info(json_data):
80
  "SD 2.1": "stabilityai/stable-diffusion-2-1-base",
81
  "SD 2.0": "stabilityai/stable-diffusion-2-base",
82
  "SD 2.1 768": "stabilityai/stable-diffusion-2-1",
83
- "SD 2.0 768": "stabilityai/stable-diffusion-2"
 
 
84
  }
85
  base_model = model_mapping[model_version["baseModel"]]
86
  info = {
 
80
  "SD 2.1": "stabilityai/stable-diffusion-2-1-base",
81
  "SD 2.0": "stabilityai/stable-diffusion-2-base",
82
  "SD 2.1 768": "stabilityai/stable-diffusion-2-1",
83
+ "SD 2.0 768": "stabilityai/stable-diffusion-2",
84
+ "Flux.1 D": "black-forest-labs/FLUX.1-dev",
85
+ "Flux.1 S": "black-forest-labs/FLUX.1-schnell"
86
  }
87
  base_model = model_mapping[model_version["baseModel"]]
88
  info = {