Spaces:
Running
on
Zero
Running
on
Zero
RageshAntony
commited on
added all models
Browse files- check_app.py +28 -4
check_app.py
CHANGED
@@ -19,17 +19,41 @@ MODEL_CONFIGS = {
|
|
19 |
"FLUX": {
|
20 |
"repo_id": "black-forest-labs/FLUX.1-dev",
|
21 |
"pipeline_class": FluxPipeline,
|
22 |
-
"cache_dir" : cache_dir
|
23 |
},
|
24 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
"repo_id": "fal/AuraFlow",
|
26 |
"pipeline_class": AuraFlowPipeline,
|
27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
28 |
},
|
29 |
"Lumina": {
|
30 |
"repo_id": "Alpha-VLLM/Lumina-Next-SFT-diffusers",
|
31 |
"pipeline_class": LuminaText2ImgPipeline,
|
32 |
-
|
33 |
}
|
34 |
}
|
35 |
|
|
|
19 |
"FLUX": {
|
20 |
"repo_id": "black-forest-labs/FLUX.1-dev",
|
21 |
"pipeline_class": FluxPipeline,
|
|
|
22 |
},
|
23 |
+
"Stable Diffusion 3.5": {
|
24 |
+
"repo_id": "stabilityai/stable-diffusion-3.5-large",
|
25 |
+
"pipeline_class": StableDiffusion3Pipeline,
|
26 |
+
|
27 |
+
},
|
28 |
+
"PixArt": {
|
29 |
+
"repo_id": "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS",
|
30 |
+
"pipeline_class": PixArtSigmaPipeline,
|
31 |
+
|
32 |
+
},
|
33 |
+
"SANA": {
|
34 |
+
"repo_id": "Efficient-Large-Model/Sana_1600M_1024px_BF16_diffusers",
|
35 |
+
"pipeline_class": SanaPipeline,
|
36 |
+
|
37 |
+
},
|
38 |
+
"AuraFlow": {
|
39 |
"repo_id": "fal/AuraFlow",
|
40 |
"pipeline_class": AuraFlowPipeline,
|
41 |
+
|
42 |
+
},
|
43 |
+
"Kandinsky": {
|
44 |
+
"repo_id": "kandinsky-community/kandinsky-3",
|
45 |
+
"pipeline_class": Kandinsky3Pipeline,
|
46 |
+
|
47 |
+
},
|
48 |
+
"Hunyuan": {
|
49 |
+
"repo_id": "Tencent-Hunyuan/HunyuanDiT-Diffusers",
|
50 |
+
"pipeline_class": HunyuanDiTPipeline,
|
51 |
+
|
52 |
},
|
53 |
"Lumina": {
|
54 |
"repo_id": "Alpha-VLLM/Lumina-Next-SFT-diffusers",
|
55 |
"pipeline_class": LuminaText2ImgPipeline,
|
56 |
+
|
57 |
}
|
58 |
}
|
59 |
|