Hugging Face's logo

Spaces:
YGu1998
/
sid-dit
Sleeping

App Files Community
YGu1998 commited on
Commit
2672f99
·
verified ·
1 Parent(s): b82506f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -20
app.py CHANGED
@@ -33,30 +33,20 @@ torch_dtype = torch.float16
33
  # }
34
 
35
  MODEL_OPTIONS = {
36
- # SD3 / SD3.5
37
- "SiD-Flow-SD3-medium": "YGu1998/SiD-DiT-SD3-medium",
38
- "SiDA-Flow-SD3-medium": "YGu1998/SiD-alpha-DiT-SD3-medium",
39
-
40
- "SiD-Flow-SD3.5-medium": "YGu1998/SiD-DiT-SD3.5-medium",
41
- "SiDA-Flow-SD3.5-medium": "YGu1998/SiD-alpha-DiT-SD3.5-medium",
42
- "SiD-Flow-SD3.5-large": "YGu1998/SiD-DiT-SD3.5-large",
43
- "SiDA-Flow-SD3.5-large": "YGu1998/SiD-alpha-DiT-SD3.5-large",
44
 
45
  # SANA rectified-flow (non-Sprint)
46
- "SiD-Flow-Sana-0.6B-512-res": "YGu1998/SiD-DiT-SANA-0.6B-RectifiedFlow",
47
- "SiDA-Flow-Sana-0.6B-512-res": "YGu1998/SiD-alpha-DiT-SANA-0.6B-RectifiedFlow",
48
- "SiD-Flow-Sana-1.6B-512-res": "YGu1998/SiD-DiT-SANA-1.6B-RectifiedFlow",
49
- "SiDA-Flow-Sana-1.6B-512-res": "YGu1998/SiD-alpha-DiT-SANA-1.6B-RectifiedFlow",
50
 
51
  # SANA Sprint (TrigFlow)
52
- "SiD-Flow-Sana-Sprint-0.6B-1024-res": "YGu1998/SiD-DiT-SANA-0.6B-TrigFlow",
53
- "SiDA-Flow-Sana-Sprint-0.6B-1024-res": "YGu1998/SiD-alpha-DiT-SANA-0.6B-TrigFlow",
54
- "SiD-Flow-Sana-Sprint-1.6B-1024-res": "YGu1998/SiD-DiT-SANA-1.6B-TrigFlow",
55
- "SiDA-Flow-Sana-Sprint-1.6B-1024-res": "YGu1998/SiD-alpha-DiT-SANA-1.6B-TrigFlow",
56
-
57
- # FLUX
58
- "SiD-Flow-Flux-1024-res": "YGu1998/SiD-DiT-Flux-1024-res",
59
- "SiD-Flow-Flux-512-res": "YGu1998/SiD-DiT-Flux-512-res",
60
  }
61
 
62
 
 
33
  # }
34
 
35
  MODEL_OPTIONS = {
36
+
 
 
 
 
 
 
 
37
 
38
  # SANA rectified-flow (non-Sprint)
39
+ "SiD-DiT-SANA-0.6B-RectifiedFlow": "YGu1998/SiD-DiT-SANA-0.6B-RectifiedFlow",
40
+ "SiD-alpha-DiT-SANA-0.6B-RectifiedFlow": "YGu1998/SiD-alpha-DiT-SANA-0.6B-RectifiedFlow",
41
+ "YGu1998/SiD-DiT-SANA-1.6B-RectifiedFlow": "YGu1998/SiD-DiT-SANA-1.6B-RectifiedFlow",
42
+ "YGu1998/SiD-alpha-DiT-SANA-1.6B-RectifiedFlow": "YGu1998/SiD-alpha-DiT-SANA-1.6B-RectifiedFlow",
43
 
44
  # SANA Sprint (TrigFlow)
45
+ "SiD-DiT-SANA-0.6B-TrigFlow": "YGu1998/SiD-DiT-SANA-0.6B-TrigFlow",
46
+ "SiD-alpha-DiT-SANA-0.6B-TrigFlow": "YGu1998/SiD-alpha-DiT-SANA-0.6B-TrigFlow",
47
+ "SiD-DiT-SANA-1.6B-TrigFlow": "YGu1998/SiD-DiT-SANA-1.6B-TrigFlow",
48
+ "SiD-alpha-DiT-SANA-1.6B-TrigFlow": "YGu1998/SiD-alpha-DiT-SANA-1.6B-TrigFlow",
49
+
 
 
 
50
  }
51
 
52