{
  "metadata": {
    "total_size": 7759609856
  },
  "weight_map": {
    "lm_head.weight": "model-00002-of-00002.safetensors",
    "model.embed_tokens.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_k_0.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_k_1.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_k_2.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_k_final.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_o_0.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_o_1.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_o_2.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_o_final.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_q_0.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_q_1.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_q_2.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_q_final.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_v_0.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_v_1.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_v_2.weight": "model-00001-of-00002.safetensors",
    "model.layers.0.self_attn.custom_attn.W_v_final.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.input_layernorm.weight": "model-00002-of-00002.safetensors",
    "model.layers.1.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
    "model.layers.1.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
    "model.layers.1.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
    "model.layers.1.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_k_0.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_k_1.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_k_2.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_k_final.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_o_0.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_o_1.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_o_2.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_o_final.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_q_0.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_q_1.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_q_2.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_q_final.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_v_0.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_v_1.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_v_2.weight": "model-00001-of-00002.safetensors",
    "model.layers.1.self_attn.custom_attn.W_v_final.weight": "model-00001-of-00002.safetensors",
    "model.norm.weight": "model-00002-of-00002.safetensors"
  }
}