RageshAntony commited on
Commit
765f56e
·
verified ·
1 Parent(s): 9ae1da0

kand & luma

Browse files
Files changed (1) hide show
  1. app.py +0 -30
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,