Baekpica commited on
Commit
cd504d0
·
verified ·
1 Parent(s): bdb8759

Add Solar architecture contract

Browse files
Files changed (1) hide show
  1. architecture.json +198 -0
architecture.json ADDED
@@ -0,0 +1,198 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "solar_open2",
3
+ "buffer_values": 15360,
4
+ "config_sha256": "fb6428ba165af1ace1d98f9170f6bafce061347593a94bd16b4b8aa3d6fe09f9",
5
+ "family_tensor_counts": {
6
+ "gqa_projection": 60,
7
+ "kda_matrix_projection": 324,
8
+ "kda_small_control_norm_conv": 216,
9
+ "lm_head": 1,
10
+ "norm": 97,
11
+ "routed_expert_down": 15360,
12
+ "routed_expert_gate": 15360,
13
+ "routed_expert_up": 15360,
14
+ "router_control": 96,
15
+ "shared_expert": 144,
16
+ "token_embedding": 1
17
+ },
18
+ "gqa": {
19
+ "head_dim": 128,
20
+ "kv_heads": 8,
21
+ "output_gate": true,
22
+ "query_heads": 64,
23
+ "use_rope": false
24
+ },
25
+ "gqa_layers": [
26
+ 0,
27
+ 4,
28
+ 8,
29
+ 12,
30
+ 16,
31
+ 20,
32
+ 24,
33
+ 28,
34
+ 32,
35
+ 36,
36
+ 40,
37
+ 44
38
+ ],
39
+ "kda": {
40
+ "allow_negative_eigenvalue": true,
41
+ "gate_equation": "g=-exp(A_log)*softplus(g_raw+dt_bias), clamped at -5",
42
+ "head_dim": 128,
43
+ "heads": 64,
44
+ "short_conv_kernel": 4,
45
+ "source": "upstageAI/transformers modular_solar_open2.py",
46
+ "state_equation": "S_t=exp(g_t)*S_(t-1)+k_t*(beta_t*(v_t-S_decay^T*k_t))^T",
47
+ "use_full_proj": false
48
+ },
49
+ "kda_layers": [
50
+ 1,
51
+ 2,
52
+ 3,
53
+ 5,
54
+ 6,
55
+ 7,
56
+ 9,
57
+ 10,
58
+ 11,
59
+ 13,
60
+ 14,
61
+ 15,
62
+ 17,
63
+ 18,
64
+ 19,
65
+ 21,
66
+ 22,
67
+ 23,
68
+ 25,
69
+ 26,
70
+ 27,
71
+ 29,
72
+ 30,
73
+ 31,
74
+ 33,
75
+ 34,
76
+ 35,
77
+ 37,
78
+ 38,
79
+ 39,
80
+ 41,
81
+ 42,
82
+ 43,
83
+ 45,
84
+ 46,
85
+ 47
86
+ ],
87
+ "layer_schedule": [
88
+ "GQA",
89
+ "KDA",
90
+ "KDA",
91
+ "KDA",
92
+ "GQA",
93
+ "KDA",
94
+ "KDA",
95
+ "KDA",
96
+ "GQA",
97
+ "KDA",
98
+ "KDA",
99
+ "KDA",
100
+ "GQA",
101
+ "KDA",
102
+ "KDA",
103
+ "KDA",
104
+ "GQA",
105
+ "KDA",
106
+ "KDA",
107
+ "KDA",
108
+ "GQA",
109
+ "KDA",
110
+ "KDA",
111
+ "KDA",
112
+ "GQA",
113
+ "KDA",
114
+ "KDA",
115
+ "KDA",
116
+ "GQA",
117
+ "KDA",
118
+ "KDA",
119
+ "KDA",
120
+ "GQA",
121
+ "KDA",
122
+ "KDA",
123
+ "KDA",
124
+ "GQA",
125
+ "KDA",
126
+ "KDA",
127
+ "KDA",
128
+ "GQA",
129
+ "KDA",
130
+ "KDA",
131
+ "KDA",
132
+ "GQA",
133
+ "KDA",
134
+ "KDA",
135
+ "KDA"
136
+ ],
137
+ "layers": 48,
138
+ "moe": {
139
+ "normalize_topk_weights": true,
140
+ "routed_experts": 320,
141
+ "routed_scaling_factor": 1.0,
142
+ "routed_width": 1280,
143
+ "router_activation": "sigmoid",
144
+ "shared_experts": 1,
145
+ "top_k": 8
146
+ },
147
+ "normalized_tensor_patterns": {
148
+ "lm_head.weight": 1,
149
+ "model.embed_tokens.weight": 1,
150
+ "model.layers.{layer}.input_layernorm.weight": 48,
151
+ "model.layers.{layer}.mlp.experts.{expert}.down_proj.weight": 15360,
152
+ "model.layers.{layer}.mlp.experts.{expert}.gate_proj.weight": 15360,
153
+ "model.layers.{layer}.mlp.experts.{expert}.up_proj.weight": 15360,
154
+ "model.layers.{layer}.mlp.gate.e_score_correction_bias": 48,
155
+ "model.layers.{layer}.mlp.gate.weight": 48,
156
+ "model.layers.{layer}.mlp.shared_experts.down_proj.weight": 48,
157
+ "model.layers.{layer}.mlp.shared_experts.gate_proj.weight": 48,
158
+ "model.layers.{layer}.mlp.shared_experts.up_proj.weight": 48,
159
+ "model.layers.{layer}.post_attention_layernorm.weight": 48,
160
+ "model.layers.{layer}.self_attn.A_log": 36,
161
+ "model.layers.{layer}.self_attn.b_proj.weight": 36,
162
+ "model.layers.{layer}.self_attn.dt_bias": 36,
163
+ "model.layers.{layer}.self_attn.f_a_proj.weight": 36,
164
+ "model.layers.{layer}.self_attn.f_b_proj.weight": 36,
165
+ "model.layers.{layer}.self_attn.g_a_proj.weight": 36,
166
+ "model.layers.{layer}.self_attn.g_b_proj.weight": 36,
167
+ "model.layers.{layer}.self_attn.g_proj.weight": 12,
168
+ "model.layers.{layer}.self_attn.k_conv1d.weight": 36,
169
+ "model.layers.{layer}.self_attn.k_proj.weight": 48,
170
+ "model.layers.{layer}.self_attn.o_norm.weight": 36,
171
+ "model.layers.{layer}.self_attn.o_proj.weight": 48,
172
+ "model.layers.{layer}.self_attn.q_conv1d.weight": 36,
173
+ "model.layers.{layer}.self_attn.q_proj.weight": 48,
174
+ "model.layers.{layer}.self_attn.v_conv1d.weight": 36,
175
+ "model.layers.{layer}.self_attn.v_proj.weight": 48,
176
+ "model.norm.weight": 1
177
+ },
178
+ "schema_version": 1,
179
+ "source": {
180
+ "repo": "upstage/Solar-Open2-250B",
181
+ "revision": "9190fbe63a2ad8e17fc766ccceb36de7c66f004b"
182
+ },
183
+ "source_dtype_counts": {
184
+ "BF16": 46971,
185
+ "F32": 48
186
+ },
187
+ "stored_values": 250287810304,
188
+ "tensor_count": 47019,
189
+ "tensor_header_sha256": "97281a0595ceeeec8c7579b8a384b9d5c35a4e5956e16b1b96cb690bd63c7f58",
190
+ "trainable_parameters": 250287794944,
191
+ "validation": {
192
+ "errors": [],
193
+ "ok": true,
194
+ "warnings": [
195
+ "pilot projection 97.76 GiB exceeds the 92 GiB artifact gate; pilot is an inventory/loadability artifact and must not be published as the capacity candidate"
196
+ ]
197
+ }
198
+ }