Spaces:
Running
on
Zero
Running
on
Zero
demo kand
Browse files
app.py
CHANGED
@@ -33,46 +33,12 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
|
|
33 |
|
34 |
# Model configurations
|
35 |
MODEL_CONFIGS = {
|
36 |
-
|
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,
|
74 |
-
#"cache_dir" : cache_dir
|
75 |
-
}
|
76 |
|
77 |
|
78 |
}
|
|
|
33 |
|
34 |
# Model configurations
|
35 |
MODEL_CONFIGS = {
|
36 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
"Kandinsky": {
|
38 |
"repo_id": "kandinsky-community/kandinsky-3",
|
39 |
"pipeline_class": Kandinsky3Pipeline,
|
40 |
#"cache_dir" : cache_dir
|
41 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
42 |
|
43 |
|
44 |
}
|