diff --git "a/v1c_full_upscale_report.json" "b/v1c_full_upscale_report.json" new file mode 100644--- /dev/null +++ "b/v1c_full_upscale_report.json" @@ -0,0 +1,7813 @@ +{ + "changed_tensors": { + "lm_head.weight": { + "old_shape": [ + 154880, + 1024 + ], + "new_shape": [ + 154880, + 6144 + ] + }, + "model.embed_tokens.weight": { + "old_shape": [ + 154880, + 1024 + ], + "new_shape": [ + 154880, + 6144 + ] + }, + "model.layers.0.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.0.mlp.down_proj.weight": { + "old_shape": [ + 1024, + 4096 + ], + "new_shape": [ + 6144, + 12288 + ] + }, + "model.layers.0.mlp.gate_proj.weight": { + "old_shape": [ + 4096, + 1024 + ], + "new_shape": [ + 12288, + 6144 + ] + }, + "model.layers.0.mlp.up_proj.weight": { + "old_shape": [ + 4096, + 1024 + ], + "new_shape": [ + 12288, + 6144 + ] + }, + "model.layers.0.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.0.self_attn.indexer.k_norm.bias": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.0.self_attn.indexer.k_norm.weight": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.0.self_attn.indexer.weights_proj.weight": { + "old_shape": [ + 8, + 1024 + ], + "new_shape": [ + 32, + 6144 + ] + }, + "model.layers.0.self_attn.indexer.wk.weight": { + "old_shape": [ + 64, + 1024 + ], + "new_shape": [ + 128, + 6144 + ] + }, + "model.layers.0.self_attn.indexer.wq_b.weight": { + "old_shape": [ + 512, + 512 + ], + "new_shape": [ + 4096, + 2048 + ] + }, + "model.layers.0.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.0.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.0.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.0.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.0.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.0.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.0.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.1.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.1.mlp.down_proj.weight": { + "old_shape": [ + 1024, + 4096 + ], + "new_shape": [ + 6144, + 12288 + ] + }, + "model.layers.1.mlp.gate_proj.weight": { + "old_shape": [ + 4096, + 1024 + ], + "new_shape": [ + 12288, + 6144 + ] + }, + "model.layers.1.mlp.up_proj.weight": { + "old_shape": [ + 4096, + 1024 + ], + "new_shape": [ + 12288, + 6144 + ] + }, + "model.layers.1.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.1.self_attn.indexer.k_norm.bias": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.1.self_attn.indexer.k_norm.weight": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.1.self_attn.indexer.weights_proj.weight": { + "old_shape": [ + 8, + 1024 + ], + "new_shape": [ + 32, + 6144 + ] + }, + "model.layers.1.self_attn.indexer.wk.weight": { + "old_shape": [ + 64, + 1024 + ], + "new_shape": [ + 128, + 6144 + ] + }, + "model.layers.1.self_attn.indexer.wq_b.weight": { + "old_shape": [ + 512, + 512 + ], + "new_shape": [ + 4096, + 2048 + ] + }, + "model.layers.1.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.1.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.1.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.1.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.1.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.1.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.1.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.10.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.10.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.10.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.10.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.10.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.10.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.10.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.10.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.10.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.10.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.10.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.10.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.11.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.11.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.11.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.11.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.11.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.11.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.11.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.11.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.11.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.11.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.11.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.11.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.12.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.12.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.12.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.12.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.12.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.12.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.12.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.12.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.12.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.12.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.12.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.12.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.13.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.13.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.13.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.13.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.13.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.13.self_attn.indexer.k_norm.bias": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.13.self_attn.indexer.k_norm.weight": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.13.self_attn.indexer.weights_proj.weight": { + "old_shape": [ + 8, + 1024 + ], + "new_shape": [ + 32, + 6144 + ] + }, + "model.layers.13.self_attn.indexer.wk.weight": { + "old_shape": [ + 64, + 1024 + ], + "new_shape": [ + 128, + 6144 + ] + }, + "model.layers.13.self_attn.indexer.wq_b.weight": { + "old_shape": [ + 512, + 512 + ], + "new_shape": [ + 4096, + 2048 + ] + }, + "model.layers.13.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.13.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.13.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.13.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.13.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.13.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.13.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.2.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.2.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.2.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.2.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.2.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.2.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.2.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.2.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.2.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.2.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.2.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.2.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.3.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.3.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.3.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.3.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.3.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.3.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.3.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.3.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.3.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.3.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.3.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.3.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.4.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.4.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.4.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.4.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.4.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.4.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.4.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.4.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.4.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.4.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.4.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.4.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.5.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.5.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.5.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.5.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.5.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.5.self_attn.indexer.k_norm.bias": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.5.self_attn.indexer.k_norm.weight": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.5.self_attn.indexer.weights_proj.weight": { + "old_shape": [ + 8, + 1024 + ], + "new_shape": [ + 32, + 6144 + ] + }, + "model.layers.5.self_attn.indexer.wk.weight": { + "old_shape": [ + 64, + 1024 + ], + "new_shape": [ + 128, + 6144 + ] + }, + "model.layers.5.self_attn.indexer.wq_b.weight": { + "old_shape": [ + 512, + 512 + ], + "new_shape": [ + 4096, + 2048 + ] + }, + "model.layers.5.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.5.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.5.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.5.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.5.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.5.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.5.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.6.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.6.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.6.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.6.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.6.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.6.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.6.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.6.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.6.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.6.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.6.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.6.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.7.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.7.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.7.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.7.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.7.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.7.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.7.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.7.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.7.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.7.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.7.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.7.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.8.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.8.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.8.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.8.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.8.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.8.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.8.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.8.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.8.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.8.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.8.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.8.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.layers.9.input_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.9.mlp.experts.0.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.0.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.0.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.1.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.1.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.1.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.10.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.10.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.10.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.11.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.11.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.11.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.12.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.12.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.12.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.13.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.13.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.13.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.14.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.14.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.14.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.15.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.15.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.15.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.2.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.2.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.2.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.3.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.3.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.3.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.4.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.4.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.4.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.5.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.5.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.5.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.6.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.6.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.6.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.7.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.7.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.7.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.8.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.8.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.8.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.9.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.experts.9.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.experts.9.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.gate.e_score_correction_bias": { + "old_shape": [ + 16 + ], + "new_shape": [ + 256 + ] + }, + "model.layers.9.mlp.gate.weight": { + "old_shape": [ + 16, + 1024 + ], + "new_shape": [ + 256, + 6144 + ] + }, + "model.layers.9.mlp.shared_experts.down_proj.weight": { + "old_shape": [ + 1024, + 512 + ], + "new_shape": [ + 6144, + 2048 + ] + }, + "model.layers.9.mlp.shared_experts.gate_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.mlp.shared_experts.up_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.post_attention_layernorm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + }, + "model.layers.9.self_attn.indexer.k_norm.bias": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.9.self_attn.indexer.k_norm.weight": { + "old_shape": [ + 64 + ], + "new_shape": [ + 128 + ] + }, + "model.layers.9.self_attn.indexer.weights_proj.weight": { + "old_shape": [ + 8, + 1024 + ], + "new_shape": [ + 32, + 6144 + ] + }, + "model.layers.9.self_attn.indexer.wk.weight": { + "old_shape": [ + 64, + 1024 + ], + "new_shape": [ + 128, + 6144 + ] + }, + "model.layers.9.self_attn.indexer.wq_b.weight": { + "old_shape": [ + 512, + 512 + ], + "new_shape": [ + 4096, + 2048 + ] + }, + "model.layers.9.self_attn.kv_a_layernorm.weight": { + "old_shape": [ + 128 + ], + "new_shape": [ + 512 + ] + }, + "model.layers.9.self_attn.kv_a_proj_with_mqa.weight": { + "old_shape": [ + 160, + 1024 + ], + "new_shape": [ + 576, + 6144 + ] + }, + "model.layers.9.self_attn.kv_b_proj.weight": { + "old_shape": [ + 3584, + 128 + ], + "new_shape": [ + 28672, + 512 + ] + }, + "model.layers.9.self_attn.o_proj.weight": { + "old_shape": [ + 1024, + 2048 + ], + "new_shape": [ + 6144, + 16384 + ] + }, + "model.layers.9.self_attn.q_a_layernorm.weight": { + "old_shape": [ + 512 + ], + "new_shape": [ + 2048 + ] + }, + "model.layers.9.self_attn.q_a_proj.weight": { + "old_shape": [ + 512, + 1024 + ], + "new_shape": [ + 2048, + 6144 + ] + }, + "model.layers.9.self_attn.q_b_proj.weight": { + "old_shape": [ + 2048, + 512 + ], + "new_shape": [ + 16384, + 2048 + ] + }, + "model.norm.weight": { + "old_shape": [ + 1024 + ], + "new_shape": [ + 6144 + ] + } + }, + "missing_experts_added": 8640, + "missing_shared_indexers_added": 45, + "source_checkpoint": "/root/inference-v2/mini-glm-5.2/v1a-mini-model/checkpoints/stage-b2-finewebedu16k-cont100m-2gpu/final", + "output_checkpoint": "/root/inference-v2/mini-glm-5.2/v1c-upscale/checkpoints/b2-final-full-original-dims-tiled-sglang", + "target_total_size_bytes": 242467656704, + "num_shards": 57, + "num_changed_tensors": 796 +}