Spaces:
Runtime error
Runtime error
kand & luma
Browse files
app.py
CHANGED
|
@@ -33,41 +33,11 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
|
|
| 33 |
|
| 34 |
# Model configurations
|
| 35 |
MODEL_CONFIGS = {
|
| 36 |
-
"FLUX": {
|
| 37 |
-
"repo_id": "black-forest-labs/FLUX.1-dev",
|
| 38 |
-
"pipeline_class": FluxPipeline,
|
| 39 |
-
#"cache_dir" : cache_dir
|
| 40 |
-
},
|
| 41 |
-
"Stable Diffusion 3.5": {
|
| 42 |
-
"repo_id": "stabilityai/stable-diffusion-3.5-large",
|
| 43 |
-
"pipeline_class": StableDiffusion3Pipeline,
|
| 44 |
-
#"cache_dir" : cache_dir
|
| 45 |
-
},
|
| 46 |
-
"PixArt": {
|
| 47 |
-
"repo_id": "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS",
|
| 48 |
-
"pipeline_class": PixArtSigmaPipeline,
|
| 49 |
-
#"cache_dir" : cache_dir
|
| 50 |
-
},
|
| 51 |
-
"SANA": {
|
| 52 |
-
"repo_id": "Efficient-Large-Model/Sana_1600M_1024px_BF16_diffusers",
|
| 53 |
-
"pipeline_class": SanaPipeline,
|
| 54 |
-
#"cache_dir" : cache_dir
|
| 55 |
-
},
|
| 56 |
-
"AuraFlow": {
|
| 57 |
-
"repo_id": "fal/AuraFlow",
|
| 58 |
-
"pipeline_class": AuraFlowPipeline,
|
| 59 |
-
#"cache_dir" : cache_dir
|
| 60 |
-
},
|
| 61 |
"Kandinsky": {
|
| 62 |
"repo_id": "kandinsky-community/kandinsky-3",
|
| 63 |
"pipeline_class": Kandinsky3Pipeline,
|
| 64 |
#"cache_dir" : cache_dir
|
| 65 |
},
|
| 66 |
-
"Hunyuan": {
|
| 67 |
-
"repo_id": "Tencent-Hunyuan/HunyuanDiT-Diffusers",
|
| 68 |
-
"pipeline_class": HunyuanDiTPipeline,
|
| 69 |
-
#"cache_dir" : cache_dir
|
| 70 |
-
},
|
| 71 |
"Lumina": {
|
| 72 |
"repo_id": "Alpha-VLLM/Lumina-Next-SFT-diffusers",
|
| 73 |
"pipeline_class": LuminaText2ImgPipeline,
|
|
|
|
| 33 |
|
| 34 |
# Model configurations
|
| 35 |
MODEL_CONFIGS = {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
"Kandinsky": {
|
| 37 |
"repo_id": "kandinsky-community/kandinsky-3",
|
| 38 |
"pipeline_class": Kandinsky3Pipeline,
|
| 39 |
#"cache_dir" : cache_dir
|
| 40 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
"Lumina": {
|
| 42 |
"repo_id": "Alpha-VLLM/Lumina-Next-SFT-diffusers",
|
| 43 |
"pipeline_class": LuminaText2ImgPipeline,
|