{
  "metadata": {
    "total_size": 5711806464
  },
  "weight_map": {
    "embed_out.weight": "model-00002-of-00002.safetensors",
    "gpt_neox.embed_in.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.final_layer_norm.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.final_layer_norm.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.attention.dense.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.attention.dense.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.attention.query_key_value.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.attention.query_key_value.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.input_layernorm.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.mlp.dense_4h_to_h.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.mlp.dense_4h_to_h.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.mlp.dense_h_to_4h.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.mlp.dense_h_to_4h.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.post_attention_layernorm.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.attention.dense.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.attention.dense.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.attention.query_key_value.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.attention.query_key_value.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.input_layernorm.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.mlp.dense_4h_to_h.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.mlp.dense_4h_to_h.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.mlp.dense_h_to_4h.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.mlp.dense_h_to_4h.weight": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.post_attention_layernorm.bias": "model-00001-of-00002.safetensors",
    "gpt_neox.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors"
  }
}