Spaces:
Running
on
Zero
Running
on
Zero
flux to first
Browse files
app.py
CHANGED
|
@@ -29,14 +29,15 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
|
|
| 29 |
|
| 30 |
# Model configurations
|
| 31 |
MODEL_CONFIGS = {
|
| 32 |
-
"Stable Diffusion 3.5": {
|
| 33 |
-
"repo_id": "stabilityai/stable-diffusion-3.5-large",
|
| 34 |
-
"pipeline_class": StableDiffusion3Pipeline
|
| 35 |
-
},
|
| 36 |
"FLUX": {
|
| 37 |
"repo_id": "black-forest-labs/FLUX.1-dev",
|
| 38 |
"pipeline_class": FluxPipeline
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
}
|
|
|
|
| 40 |
}
|
| 41 |
|
| 42 |
# Dictionary to store model pipelines
|
|
|
|
| 29 |
|
| 30 |
# Model configurations
|
| 31 |
MODEL_CONFIGS = {
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
"FLUX": {
|
| 33 |
"repo_id": "black-forest-labs/FLUX.1-dev",
|
| 34 |
"pipeline_class": FluxPipeline
|
| 35 |
+
},
|
| 36 |
+
"Stable Diffusion 3.5": {
|
| 37 |
+
"repo_id": "stabilityai/stable-diffusion-3.5-large",
|
| 38 |
+
"pipeline_class": StableDiffusion3Pipeline
|
| 39 |
}
|
| 40 |
+
|
| 41 |
}
|
| 42 |
|
| 43 |
# Dictionary to store model pipelines
|