RageshAntony commited on
Commit
40e3650
·
verified ·
1 Parent(s): cc32d9c

added all models

Browse files
Files changed (1) hide show
  1. 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
- "AuraFlow": {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  "repo_id": "fal/AuraFlow",
26
  "pipeline_class": AuraFlowPipeline,
27
- "cache_dir": cache_dir,
 
 
 
 
 
 
 
 
 
 
28
  },
29
  "Lumina": {
30
  "repo_id": "Alpha-VLLM/Lumina-Next-SFT-diffusers",
31
  "pipeline_class": LuminaText2ImgPipeline,
32
- "cache_dir": cache_dir,
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