File size: 17,903 Bytes
0e5f2a4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
{
  "model.embed_tokens.weight": "embed.weight",
  "model.hc_head.hc_base": "hc_head_base",
  "model.hc_head.hc_fn": "hc_head_fn",
  "model.hc_head.hc_scale": "hc_head_scale",
  "model.layers.0.attn_hc.base": "layers.0.hc_attn_base",
  "model.layers.0.attn_hc.fn": "layers.0.hc_attn_fn",
  "model.layers.0.attn_hc.scale": "layers.0.hc_attn_scale",
  "model.layers.0.ffn_hc.base": "layers.0.hc_ffn_base",
  "model.layers.0.ffn_hc.fn": "layers.0.hc_ffn_fn",
  "model.layers.0.ffn_hc.scale": "layers.0.hc_ffn_scale",
  "model.layers.0.input_layernorm.weight": "layers.0.attn_norm.weight",
  "model.layers.0.mlp.experts.0.w1.weight": "layers.0.ffn.experts.0.w1.weight",
  "model.layers.0.mlp.experts.0.w2.weight": "layers.0.ffn.experts.0.w2.weight",
  "model.layers.0.mlp.experts.0.w3.weight": "layers.0.ffn.experts.0.w3.weight",
  "model.layers.0.mlp.experts.1.w1.weight": "layers.0.ffn.experts.1.w1.weight",
  "model.layers.0.mlp.experts.1.w2.weight": "layers.0.ffn.experts.1.w2.weight",
  "model.layers.0.mlp.experts.1.w3.weight": "layers.0.ffn.experts.1.w3.weight",
  "model.layers.0.mlp.experts.10.w1.weight": "layers.0.ffn.experts.10.w1.weight",
  "model.layers.0.mlp.experts.10.w2.weight": "layers.0.ffn.experts.10.w2.weight",
  "model.layers.0.mlp.experts.10.w3.weight": "layers.0.ffn.experts.10.w3.weight",
  "model.layers.0.mlp.experts.11.w1.weight": "layers.0.ffn.experts.11.w1.weight",
  "model.layers.0.mlp.experts.11.w2.weight": "layers.0.ffn.experts.11.w2.weight",
  "model.layers.0.mlp.experts.11.w3.weight": "layers.0.ffn.experts.11.w3.weight",
  "model.layers.0.mlp.experts.12.w1.weight": "layers.0.ffn.experts.12.w1.weight",
  "model.layers.0.mlp.experts.12.w2.weight": "layers.0.ffn.experts.12.w2.weight",
  "model.layers.0.mlp.experts.12.w3.weight": "layers.0.ffn.experts.12.w3.weight",
  "model.layers.0.mlp.experts.13.w1.weight": "layers.0.ffn.experts.13.w1.weight",
  "model.layers.0.mlp.experts.13.w2.weight": "layers.0.ffn.experts.13.w2.weight",
  "model.layers.0.mlp.experts.13.w3.weight": "layers.0.ffn.experts.13.w3.weight",
  "model.layers.0.mlp.experts.14.w1.weight": "layers.0.ffn.experts.14.w1.weight",
  "model.layers.0.mlp.experts.14.w2.weight": "layers.0.ffn.experts.14.w2.weight",
  "model.layers.0.mlp.experts.14.w3.weight": "layers.0.ffn.experts.14.w3.weight",
  "model.layers.0.mlp.experts.15.w1.weight": "layers.0.ffn.experts.15.w1.weight",
  "model.layers.0.mlp.experts.15.w2.weight": "layers.0.ffn.experts.15.w2.weight",
  "model.layers.0.mlp.experts.15.w3.weight": "layers.0.ffn.experts.15.w3.weight",
  "model.layers.0.mlp.experts.2.w1.weight": "layers.0.ffn.experts.2.w1.weight",
  "model.layers.0.mlp.experts.2.w2.weight": "layers.0.ffn.experts.2.w2.weight",
  "model.layers.0.mlp.experts.2.w3.weight": "layers.0.ffn.experts.2.w3.weight",
  "model.layers.0.mlp.experts.3.w1.weight": "layers.0.ffn.experts.3.w1.weight",
  "model.layers.0.mlp.experts.3.w2.weight": "layers.0.ffn.experts.3.w2.weight",
  "model.layers.0.mlp.experts.3.w3.weight": "layers.0.ffn.experts.3.w3.weight",
  "model.layers.0.mlp.experts.4.w1.weight": "layers.0.ffn.experts.4.w1.weight",
  "model.layers.0.mlp.experts.4.w2.weight": "layers.0.ffn.experts.4.w2.weight",
  "model.layers.0.mlp.experts.4.w3.weight": "layers.0.ffn.experts.4.w3.weight",
  "model.layers.0.mlp.experts.5.w1.weight": "layers.0.ffn.experts.5.w1.weight",
  "model.layers.0.mlp.experts.5.w2.weight": "layers.0.ffn.experts.5.w2.weight",
  "model.layers.0.mlp.experts.5.w3.weight": "layers.0.ffn.experts.5.w3.weight",
  "model.layers.0.mlp.experts.6.w1.weight": "layers.0.ffn.experts.6.w1.weight",
  "model.layers.0.mlp.experts.6.w2.weight": "layers.0.ffn.experts.6.w2.weight",
  "model.layers.0.mlp.experts.6.w3.weight": "layers.0.ffn.experts.6.w3.weight",
  "model.layers.0.mlp.experts.7.w1.weight": "layers.0.ffn.experts.7.w1.weight",
  "model.layers.0.mlp.experts.7.w2.weight": "layers.0.ffn.experts.7.w2.weight",
  "model.layers.0.mlp.experts.7.w3.weight": "layers.0.ffn.experts.7.w3.weight",
  "model.layers.0.mlp.experts.8.w1.weight": "layers.0.ffn.experts.8.w1.weight",
  "model.layers.0.mlp.experts.8.w2.weight": "layers.0.ffn.experts.8.w2.weight",
  "model.layers.0.mlp.experts.8.w3.weight": "layers.0.ffn.experts.8.w3.weight",
  "model.layers.0.mlp.experts.9.w1.weight": "layers.0.ffn.experts.9.w1.weight",
  "model.layers.0.mlp.experts.9.w2.weight": "layers.0.ffn.experts.9.w2.weight",
  "model.layers.0.mlp.experts.9.w3.weight": "layers.0.ffn.experts.9.w3.weight",
  "model.layers.0.mlp.gate.tid2eid": "layers.0.ffn.gate.tid2eid",
  "model.layers.0.mlp.gate.weight": "layers.0.ffn.gate.weight",
  "model.layers.0.mlp.shared_experts.down_proj.weight": "layers.0.ffn.shared_experts.w2.weight",
  "model.layers.0.mlp.shared_experts.gate_proj.weight": "layers.0.ffn.shared_experts.w1.weight",
  "model.layers.0.mlp.shared_experts.up_proj.weight": "layers.0.ffn.shared_experts.w3.weight",
  "model.layers.0.post_attention_layernorm.weight": "layers.0.ffn_norm.weight",
  "model.layers.0.self_attn.compressor.gate_proj.weight": "layers.0.attn.compressor.gate_proj.weight",
  "model.layers.0.self_attn.compressor.kv_norm.weight": "layers.0.attn.compressor.kv_norm.weight",
  "model.layers.0.self_attn.compressor.kv_proj.weight": "layers.0.attn.compressor.kv_proj.weight",
  "model.layers.0.self_attn.compressor.position_bias": "layers.0.attn.compressor.position_bias",
  "model.layers.0.self_attn.kv_norm.weight": "layers.0.attn.kv_norm.weight",
  "model.layers.0.self_attn.kv_proj.weight": "layers.0.attn.wkv.weight",
  "model.layers.0.self_attn.o_a_proj.weight": "layers.0.attn.wo_a.weight",
  "model.layers.0.self_attn.o_b_proj.weight": "layers.0.attn.wo_b.weight",
  "model.layers.0.self_attn.q_a_norm.weight": "layers.0.attn.q_norm.weight",
  "model.layers.0.self_attn.q_a_proj.weight": "layers.0.attn.wq_a.weight",
  "model.layers.0.self_attn.q_b_proj.weight": "layers.0.attn.wq_b.weight",
  "model.layers.0.self_attn.sinks": "layers.0.attn.attn_sink",
  "model.layers.1.attn_hc.base": "layers.1.hc_attn_base",
  "model.layers.1.attn_hc.fn": "layers.1.hc_attn_fn",
  "model.layers.1.attn_hc.scale": "layers.1.hc_attn_scale",
  "model.layers.1.ffn_hc.base": "layers.1.hc_ffn_base",
  "model.layers.1.ffn_hc.fn": "layers.1.hc_ffn_fn",
  "model.layers.1.ffn_hc.scale": "layers.1.hc_ffn_scale",
  "model.layers.1.input_layernorm.weight": "layers.1.attn_norm.weight",
  "model.layers.1.mlp.experts.0.w1.weight": "layers.1.ffn.experts.0.w1.weight",
  "model.layers.1.mlp.experts.0.w2.weight": "layers.1.ffn.experts.0.w2.weight",
  "model.layers.1.mlp.experts.0.w3.weight": "layers.1.ffn.experts.0.w3.weight",
  "model.layers.1.mlp.experts.1.w1.weight": "layers.1.ffn.experts.1.w1.weight",
  "model.layers.1.mlp.experts.1.w2.weight": "layers.1.ffn.experts.1.w2.weight",
  "model.layers.1.mlp.experts.1.w3.weight": "layers.1.ffn.experts.1.w3.weight",
  "model.layers.1.mlp.experts.10.w1.weight": "layers.1.ffn.experts.10.w1.weight",
  "model.layers.1.mlp.experts.10.w2.weight": "layers.1.ffn.experts.10.w2.weight",
  "model.layers.1.mlp.experts.10.w3.weight": "layers.1.ffn.experts.10.w3.weight",
  "model.layers.1.mlp.experts.11.w1.weight": "layers.1.ffn.experts.11.w1.weight",
  "model.layers.1.mlp.experts.11.w2.weight": "layers.1.ffn.experts.11.w2.weight",
  "model.layers.1.mlp.experts.11.w3.weight": "layers.1.ffn.experts.11.w3.weight",
  "model.layers.1.mlp.experts.12.w1.weight": "layers.1.ffn.experts.12.w1.weight",
  "model.layers.1.mlp.experts.12.w2.weight": "layers.1.ffn.experts.12.w2.weight",
  "model.layers.1.mlp.experts.12.w3.weight": "layers.1.ffn.experts.12.w3.weight",
  "model.layers.1.mlp.experts.13.w1.weight": "layers.1.ffn.experts.13.w1.weight",
  "model.layers.1.mlp.experts.13.w2.weight": "layers.1.ffn.experts.13.w2.weight",
  "model.layers.1.mlp.experts.13.w3.weight": "layers.1.ffn.experts.13.w3.weight",
  "model.layers.1.mlp.experts.14.w1.weight": "layers.1.ffn.experts.14.w1.weight",
  "model.layers.1.mlp.experts.14.w2.weight": "layers.1.ffn.experts.14.w2.weight",
  "model.layers.1.mlp.experts.14.w3.weight": "layers.1.ffn.experts.14.w3.weight",
  "model.layers.1.mlp.experts.15.w1.weight": "layers.1.ffn.experts.15.w1.weight",
  "model.layers.1.mlp.experts.15.w2.weight": "layers.1.ffn.experts.15.w2.weight",
  "model.layers.1.mlp.experts.15.w3.weight": "layers.1.ffn.experts.15.w3.weight",
  "model.layers.1.mlp.experts.2.w1.weight": "layers.1.ffn.experts.2.w1.weight",
  "model.layers.1.mlp.experts.2.w2.weight": "layers.1.ffn.experts.2.w2.weight",
  "model.layers.1.mlp.experts.2.w3.weight": "layers.1.ffn.experts.2.w3.weight",
  "model.layers.1.mlp.experts.3.w1.weight": "layers.1.ffn.experts.3.w1.weight",
  "model.layers.1.mlp.experts.3.w2.weight": "layers.1.ffn.experts.3.w2.weight",
  "model.layers.1.mlp.experts.3.w3.weight": "layers.1.ffn.experts.3.w3.weight",
  "model.layers.1.mlp.experts.4.w1.weight": "layers.1.ffn.experts.4.w1.weight",
  "model.layers.1.mlp.experts.4.w2.weight": "layers.1.ffn.experts.4.w2.weight",
  "model.layers.1.mlp.experts.4.w3.weight": "layers.1.ffn.experts.4.w3.weight",
  "model.layers.1.mlp.experts.5.w1.weight": "layers.1.ffn.experts.5.w1.weight",
  "model.layers.1.mlp.experts.5.w2.weight": "layers.1.ffn.experts.5.w2.weight",
  "model.layers.1.mlp.experts.5.w3.weight": "layers.1.ffn.experts.5.w3.weight",
  "model.layers.1.mlp.experts.6.w1.weight": "layers.1.ffn.experts.6.w1.weight",
  "model.layers.1.mlp.experts.6.w2.weight": "layers.1.ffn.experts.6.w2.weight",
  "model.layers.1.mlp.experts.6.w3.weight": "layers.1.ffn.experts.6.w3.weight",
  "model.layers.1.mlp.experts.7.w1.weight": "layers.1.ffn.experts.7.w1.weight",
  "model.layers.1.mlp.experts.7.w2.weight": "layers.1.ffn.experts.7.w2.weight",
  "model.layers.1.mlp.experts.7.w3.weight": "layers.1.ffn.experts.7.w3.weight",
  "model.layers.1.mlp.experts.8.w1.weight": "layers.1.ffn.experts.8.w1.weight",
  "model.layers.1.mlp.experts.8.w2.weight": "layers.1.ffn.experts.8.w2.weight",
  "model.layers.1.mlp.experts.8.w3.weight": "layers.1.ffn.experts.8.w3.weight",
  "model.layers.1.mlp.experts.9.w1.weight": "layers.1.ffn.experts.9.w1.weight",
  "model.layers.1.mlp.experts.9.w2.weight": "layers.1.ffn.experts.9.w2.weight",
  "model.layers.1.mlp.experts.9.w3.weight": "layers.1.ffn.experts.9.w3.weight",
  "model.layers.1.mlp.gate.e_score_correction_bias": "layers.1.ffn.gate.e_score_correction_bias",
  "model.layers.1.mlp.gate.weight": "layers.1.ffn.gate.weight",
  "model.layers.1.mlp.shared_experts.down_proj.weight": "layers.1.ffn.shared_experts.w2.weight",
  "model.layers.1.mlp.shared_experts.gate_proj.weight": "layers.1.ffn.shared_experts.w1.weight",
  "model.layers.1.mlp.shared_experts.up_proj.weight": "layers.1.ffn.shared_experts.w3.weight",
  "model.layers.1.post_attention_layernorm.weight": "layers.1.ffn_norm.weight",
  "model.layers.1.self_attn.compressor.gate_proj.weight": "layers.1.attn.compressor.gate_proj.weight",
  "model.layers.1.self_attn.compressor.indexer.gate_proj.weight": "layers.1.attn.compressor.indexer.gate_proj.weight",
  "model.layers.1.self_attn.compressor.indexer.kv_norm.weight": "layers.1.attn.compressor.indexer.kv_norm.weight",
  "model.layers.1.self_attn.compressor.indexer.kv_proj.weight": "layers.1.attn.compressor.indexer.kv_proj.weight",
  "model.layers.1.self_attn.compressor.indexer.position_bias": "layers.1.attn.compressor.indexer.position_bias",
  "model.layers.1.self_attn.compressor.indexer.q_b_proj.weight": "layers.1.attn.compressor.indexer.q_b_proj.weight",
  "model.layers.1.self_attn.compressor.indexer.weights_proj.weight": "layers.1.attn.compressor.indexer.weights_proj.weight",
  "model.layers.1.self_attn.compressor.kv_norm.weight": "layers.1.attn.compressor.kv_norm.weight",
  "model.layers.1.self_attn.compressor.kv_proj.weight": "layers.1.attn.compressor.kv_proj.weight",
  "model.layers.1.self_attn.compressor.position_bias": "layers.1.attn.compressor.position_bias",
  "model.layers.1.self_attn.kv_norm.weight": "layers.1.attn.kv_norm.weight",
  "model.layers.1.self_attn.kv_proj.weight": "layers.1.attn.wkv.weight",
  "model.layers.1.self_attn.o_a_proj.weight": "layers.1.attn.wo_a.weight",
  "model.layers.1.self_attn.o_b_proj.weight": "layers.1.attn.wo_b.weight",
  "model.layers.1.self_attn.q_a_norm.weight": "layers.1.attn.q_norm.weight",
  "model.layers.1.self_attn.q_a_proj.weight": "layers.1.attn.wq_a.weight",
  "model.layers.1.self_attn.q_b_proj.weight": "layers.1.attn.wq_b.weight",
  "model.layers.1.self_attn.sinks": "layers.1.attn.attn_sink",
  "model.layers.2.attn_hc.base": "layers.2.hc_attn_base",
  "model.layers.2.attn_hc.fn": "layers.2.hc_attn_fn",
  "model.layers.2.attn_hc.scale": "layers.2.hc_attn_scale",
  "model.layers.2.ffn_hc.base": "layers.2.hc_ffn_base",
  "model.layers.2.ffn_hc.fn": "layers.2.hc_ffn_fn",
  "model.layers.2.ffn_hc.scale": "layers.2.hc_ffn_scale",
  "model.layers.2.input_layernorm.weight": "layers.2.attn_norm.weight",
  "model.layers.2.mlp.experts.0.w1.weight": "layers.2.ffn.experts.0.w1.weight",
  "model.layers.2.mlp.experts.0.w2.weight": "layers.2.ffn.experts.0.w2.weight",
  "model.layers.2.mlp.experts.0.w3.weight": "layers.2.ffn.experts.0.w3.weight",
  "model.layers.2.mlp.experts.1.w1.weight": "layers.2.ffn.experts.1.w1.weight",
  "model.layers.2.mlp.experts.1.w2.weight": "layers.2.ffn.experts.1.w2.weight",
  "model.layers.2.mlp.experts.1.w3.weight": "layers.2.ffn.experts.1.w3.weight",
  "model.layers.2.mlp.experts.10.w1.weight": "layers.2.ffn.experts.10.w1.weight",
  "model.layers.2.mlp.experts.10.w2.weight": "layers.2.ffn.experts.10.w2.weight",
  "model.layers.2.mlp.experts.10.w3.weight": "layers.2.ffn.experts.10.w3.weight",
  "model.layers.2.mlp.experts.11.w1.weight": "layers.2.ffn.experts.11.w1.weight",
  "model.layers.2.mlp.experts.11.w2.weight": "layers.2.ffn.experts.11.w2.weight",
  "model.layers.2.mlp.experts.11.w3.weight": "layers.2.ffn.experts.11.w3.weight",
  "model.layers.2.mlp.experts.12.w1.weight": "layers.2.ffn.experts.12.w1.weight",
  "model.layers.2.mlp.experts.12.w2.weight": "layers.2.ffn.experts.12.w2.weight",
  "model.layers.2.mlp.experts.12.w3.weight": "layers.2.ffn.experts.12.w3.weight",
  "model.layers.2.mlp.experts.13.w1.weight": "layers.2.ffn.experts.13.w1.weight",
  "model.layers.2.mlp.experts.13.w2.weight": "layers.2.ffn.experts.13.w2.weight",
  "model.layers.2.mlp.experts.13.w3.weight": "layers.2.ffn.experts.13.w3.weight",
  "model.layers.2.mlp.experts.14.w1.weight": "layers.2.ffn.experts.14.w1.weight",
  "model.layers.2.mlp.experts.14.w2.weight": "layers.2.ffn.experts.14.w2.weight",
  "model.layers.2.mlp.experts.14.w3.weight": "layers.2.ffn.experts.14.w3.weight",
  "model.layers.2.mlp.experts.15.w1.weight": "layers.2.ffn.experts.15.w1.weight",
  "model.layers.2.mlp.experts.15.w2.weight": "layers.2.ffn.experts.15.w2.weight",
  "model.layers.2.mlp.experts.15.w3.weight": "layers.2.ffn.experts.15.w3.weight",
  "model.layers.2.mlp.experts.2.w1.weight": "layers.2.ffn.experts.2.w1.weight",
  "model.layers.2.mlp.experts.2.w2.weight": "layers.2.ffn.experts.2.w2.weight",
  "model.layers.2.mlp.experts.2.w3.weight": "layers.2.ffn.experts.2.w3.weight",
  "model.layers.2.mlp.experts.3.w1.weight": "layers.2.ffn.experts.3.w1.weight",
  "model.layers.2.mlp.experts.3.w2.weight": "layers.2.ffn.experts.3.w2.weight",
  "model.layers.2.mlp.experts.3.w3.weight": "layers.2.ffn.experts.3.w3.weight",
  "model.layers.2.mlp.experts.4.w1.weight": "layers.2.ffn.experts.4.w1.weight",
  "model.layers.2.mlp.experts.4.w2.weight": "layers.2.ffn.experts.4.w2.weight",
  "model.layers.2.mlp.experts.4.w3.weight": "layers.2.ffn.experts.4.w3.weight",
  "model.layers.2.mlp.experts.5.w1.weight": "layers.2.ffn.experts.5.w1.weight",
  "model.layers.2.mlp.experts.5.w2.weight": "layers.2.ffn.experts.5.w2.weight",
  "model.layers.2.mlp.experts.5.w3.weight": "layers.2.ffn.experts.5.w3.weight",
  "model.layers.2.mlp.experts.6.w1.weight": "layers.2.ffn.experts.6.w1.weight",
  "model.layers.2.mlp.experts.6.w2.weight": "layers.2.ffn.experts.6.w2.weight",
  "model.layers.2.mlp.experts.6.w3.weight": "layers.2.ffn.experts.6.w3.weight",
  "model.layers.2.mlp.experts.7.w1.weight": "layers.2.ffn.experts.7.w1.weight",
  "model.layers.2.mlp.experts.7.w2.weight": "layers.2.ffn.experts.7.w2.weight",
  "model.layers.2.mlp.experts.7.w3.weight": "layers.2.ffn.experts.7.w3.weight",
  "model.layers.2.mlp.experts.8.w1.weight": "layers.2.ffn.experts.8.w1.weight",
  "model.layers.2.mlp.experts.8.w2.weight": "layers.2.ffn.experts.8.w2.weight",
  "model.layers.2.mlp.experts.8.w3.weight": "layers.2.ffn.experts.8.w3.weight",
  "model.layers.2.mlp.experts.9.w1.weight": "layers.2.ffn.experts.9.w1.weight",
  "model.layers.2.mlp.experts.9.w2.weight": "layers.2.ffn.experts.9.w2.weight",
  "model.layers.2.mlp.experts.9.w3.weight": "layers.2.ffn.experts.9.w3.weight",
  "model.layers.2.mlp.gate.e_score_correction_bias": "layers.2.ffn.gate.e_score_correction_bias",
  "model.layers.2.mlp.gate.weight": "layers.2.ffn.gate.weight",
  "model.layers.2.mlp.shared_experts.down_proj.weight": "layers.2.ffn.shared_experts.w2.weight",
  "model.layers.2.mlp.shared_experts.gate_proj.weight": "layers.2.ffn.shared_experts.w1.weight",
  "model.layers.2.mlp.shared_experts.up_proj.weight": "layers.2.ffn.shared_experts.w3.weight",
  "model.layers.2.post_attention_layernorm.weight": "layers.2.ffn_norm.weight",
  "model.layers.2.self_attn.kv_norm.weight": "layers.2.attn.kv_norm.weight",
  "model.layers.2.self_attn.kv_proj.weight": "layers.2.attn.wkv.weight",
  "model.layers.2.self_attn.o_a_proj.weight": "layers.2.attn.wo_a.weight",
  "model.layers.2.self_attn.o_b_proj.weight": "layers.2.attn.wo_b.weight",
  "model.layers.2.self_attn.q_a_norm.weight": "layers.2.attn.q_norm.weight",
  "model.layers.2.self_attn.q_a_proj.weight": "layers.2.attn.wq_a.weight",
  "model.layers.2.self_attn.q_b_proj.weight": "layers.2.attn.wq_b.weight",
  "model.layers.2.self_attn.sinks": "layers.2.attn.attn_sink",
  "model.norm.weight": "norm.weight"
}