Upload BaichuanForCausalLM

config.json

@@ -0,0 +1,31 @@
+{
+  "_from_model_config": true,
+  "_name_or_path": "baichuan-inc/Baichuan2-7B-Chat",
+  "architectures": [
+    "BaichuanForCausalLM"
+  ],
+  "auto_map": {
+    "AutoConfig": "baichuan-inc/Baichuan2-7B-Chat--configuration_baichuan.BaichuanConfig",
+    "AutoModelForCausalLM": "baichuan-inc/Baichuan2-7B-Chat--modeling_baichuan.BaichuanForCausalLM"
+  },
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 4096,
+  "initializer_range": 0.02,
+  "intermediate_size": 11008,
+  "max_position_embeddings": 4096,
+  "model_max_length": 4096,
+  "model_type": "baichuan",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 32,
+  "pad_token_id": 0,
+  "rms_norm_eps": 1e-06,
+  "tie_word_embeddings": false,
+  "tokenizer_class": "BaichuanTokenizer",
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.33.0",
+  "use_cache": true,
+  "vocab_size": 125696,
+  "z_loss_weight": 0
+}

generation_config.json

@@ -0,0 +1,14 @@
+{
+  "assistant_token_id": 196,
+  "bos_token_id": 1,
+  "do_sample": true,
+  "eos_token_id": 2,
+  "max_new_tokens": 2048,
+  "pad_token_id": 0,
+  "repetition_penalty": 1.05,
+  "temperature": 0.3,
+  "top_k": 5,
+  "top_p": 0.85,
+  "transformers_version": "4.33.0",
+  "user_token_id": 195
+}

pytorch_model-00001-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:300edab21117a6c0ec39cab6a107affdc366393fa3aab9133d375fb7f3ba907f
+size 1029702570

pytorch_model-00002-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8fed74be10e2f4d3ab2e61d15f6f776488bd163e7e7a2607c48693612cd4b485
+size 943756783

pytorch_model-00003-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a415318296a149e3207d9abcdd1dc48a642669073059df91c352f2618d77208a
+size 989894127

pytorch_model-00004-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0322436fe86aae257561a7c66e6955e19a3a115b8260dd691e981bc9d26907d6
+size 899733249

pytorch_model-00005-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:06379472cd2ef590cdf25c65c5139f1876a78be6e458ef3a37bfed0be359b991
+size 943756783

pytorch_model-00006-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f66893733ede1282c698edc2a131fcdd731a6e015e135819df12b58d30b8ada0
+size 989894127

pytorch_model-00007-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1705c689df8ec0719d049f3eea082448209ca5fca3b0d1562134d0a11c8e9f9
+size 899733313

pytorch_model-00008-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:765262088d3bf7cb6d87e1a0e611005ab93cf07ff96b1d250af99c530eb1ea7e
+size 943756783

pytorch_model-00009-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0b96ba1dcbac32336fb3019ada5fca59479d47957dcf7e419e7e1fed4d294df2
+size 989894127

pytorch_model-00010-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2efabe7ccf559b72dbe146787f018e8694e36521781f7996017b97cc332c01a2
+size 899733313

pytorch_model-00011-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14c217684ed1e844f8720064114e8142c3249896bcada1a766ff9925092a676a
+size 943756783

pytorch_model-00012-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:29fe783e77dfa698fd4c73cdf93272820985c81cde5ff6d44d5fe42c0e8f6c6a
+size 989894127

pytorch_model-00013-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a0dad598606bc3d3908736020e4a53bf43b57d5d58499e2cfaa74401e174c95
+size 899733313

pytorch_model-00014-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9e41a1c0695b2a343a9e00c02d75c921630d83651851da833b0d16a4a2dc4bdd
+size 943756783

pytorch_model-00015-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:68eae088643392d48fdcc424937ec3e8f8296d2dd7bf0fbc95cf6715caaeeea3
+size 675328569

pytorch_model-00016-of-00016.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0e1f3618afc6e326936a9f7283c37b93d1ef68b9c918d7baea118b29263a0d2
+size 1029702570

pytorch_model.bin.index.json

@@ -0,0 +1,234 @@
+{
+  "metadata": {
+    "total_size": 15011946496
+  },
+  "weight_map": {
+    "lm_head.weight": "pytorch_model-00016-of-00016.bin",
+    "model.embed_tokens.weight": "pytorch_model-00001-of-00016.bin",
+    "model.layers.0.input_layernorm.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.0.mlp.down_proj.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.0.mlp.gate_proj.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.0.mlp.up_proj.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.0.post_attention_layernorm.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.0.self_attn.W_pack.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.0.self_attn.o_proj.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.1.input_layernorm.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.1.mlp.down_proj.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.1.mlp.gate_proj.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.1.mlp.up_proj.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.1.post_attention_layernorm.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.1.self_attn.W_pack.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.1.self_attn.o_proj.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.10.input_layernorm.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.10.mlp.down_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.10.mlp.gate_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.10.mlp.up_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.10.post_attention_layernorm.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.10.self_attn.W_pack.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.10.self_attn.o_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.11.input_layernorm.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.11.mlp.down_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.11.mlp.gate_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.11.mlp.up_proj.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.11.post_attention_layernorm.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.11.self_attn.W_pack.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.11.self_attn.o_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.12.input_layernorm.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.12.mlp.down_proj.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.12.mlp.gate_proj.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.12.mlp.up_proj.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.12.post_attention_layernorm.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.12.self_attn.W_pack.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.12.self_attn.o_proj.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.13.input_layernorm.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.13.mlp.down_proj.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.13.mlp.gate_proj.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.13.mlp.up_proj.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.13.post_attention_layernorm.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.13.self_attn.W_pack.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.13.self_attn.o_proj.weight": "pytorch_model-00007-of-00016.bin",
+    "model.layers.14.input_layernorm.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.14.mlp.down_proj.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.14.mlp.gate_proj.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.14.mlp.up_proj.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.14.post_attention_layernorm.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.14.self_attn.W_pack.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.14.self_attn.o_proj.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.15.input_layernorm.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.15.mlp.down_proj.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.15.mlp.gate_proj.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.15.mlp.up_proj.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.15.post_attention_layernorm.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.15.self_attn.W_pack.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.15.self_attn.o_proj.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.16.input_layernorm.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.16.mlp.down_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.16.mlp.gate_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.16.mlp.up_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.16.post_attention_layernorm.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.16.self_attn.W_pack.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.16.self_attn.o_proj.weight": "pytorch_model-00008-of-00016.bin",
+    "model.layers.17.input_layernorm.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.17.mlp.down_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.17.mlp.gate_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.17.mlp.up_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.17.post_attention_layernorm.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.17.self_attn.W_pack.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.17.self_attn.o_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.18.input_layernorm.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.18.mlp.down_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.18.mlp.gate_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.18.mlp.up_proj.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.18.post_attention_layernorm.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.18.self_attn.W_pack.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.18.self_attn.o_proj.weight": "pytorch_model-00009-of-00016.bin",
+    "model.layers.19.input_layernorm.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.19.mlp.down_proj.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.19.mlp.gate_proj.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.19.mlp.up_proj.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.19.post_attention_layernorm.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.19.self_attn.W_pack.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.19.self_attn.o_proj.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.2.input_layernorm.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.2.mlp.down_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.2.mlp.gate_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.2.mlp.up_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.2.post_attention_layernorm.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.2.self_attn.W_pack.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.2.self_attn.o_proj.weight": "pytorch_model-00002-of-00016.bin",
+    "model.layers.20.input_layernorm.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.20.mlp.down_proj.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.20.mlp.gate_proj.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.20.mlp.up_proj.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.20.post_attention_layernorm.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.20.self_attn.W_pack.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.20.self_attn.o_proj.weight": "pytorch_model-00010-of-00016.bin",
+    "model.layers.21.input_layernorm.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.21.mlp.down_proj.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.21.mlp.gate_proj.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.21.mlp.up_proj.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.21.post_attention_layernorm.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.21.self_attn.W_pack.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.21.self_attn.o_proj.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.22.input_layernorm.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.22.mlp.down_proj.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.22.mlp.gate_proj.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.22.mlp.up_proj.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.22.post_attention_layernorm.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.22.self_attn.W_pack.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.22.self_attn.o_proj.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.23.input_layernorm.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.23.mlp.down_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.23.mlp.gate_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.23.mlp.up_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.23.post_attention_layernorm.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.23.self_attn.W_pack.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.23.self_attn.o_proj.weight": "pytorch_model-00011-of-00016.bin",
+    "model.layers.24.input_layernorm.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.24.mlp.down_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.24.mlp.gate_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.24.mlp.up_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.24.post_attention_layernorm.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.24.self_attn.W_pack.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.24.self_attn.o_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.25.input_layernorm.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.25.mlp.down_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.25.mlp.gate_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.25.mlp.up_proj.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.25.post_attention_layernorm.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.25.self_attn.W_pack.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.25.self_attn.o_proj.weight": "pytorch_model-00012-of-00016.bin",
+    "model.layers.26.input_layernorm.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.26.mlp.down_proj.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.26.mlp.gate_proj.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.26.mlp.up_proj.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.26.post_attention_layernorm.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.26.self_attn.W_pack.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.26.self_attn.o_proj.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.27.input_layernorm.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.27.mlp.down_proj.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.27.mlp.gate_proj.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.27.mlp.up_proj.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.27.post_attention_layernorm.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.27.self_attn.W_pack.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.27.self_attn.o_proj.weight": "pytorch_model-00013-of-00016.bin",
+    "model.layers.28.input_layernorm.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.28.mlp.down_proj.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.28.mlp.gate_proj.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.28.mlp.up_proj.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.28.post_attention_layernorm.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.28.self_attn.W_pack.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.28.self_attn.o_proj.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.29.input_layernorm.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.29.mlp.down_proj.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.29.mlp.gate_proj.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.29.mlp.up_proj.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.29.post_attention_layernorm.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.29.self_attn.W_pack.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.29.self_attn.o_proj.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.3.input_layernorm.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.3.mlp.down_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.3.mlp.gate_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.3.mlp.up_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.3.post_attention_layernorm.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.3.self_attn.W_pack.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.3.self_attn.o_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.30.input_layernorm.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.30.mlp.down_proj.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.30.mlp.gate_proj.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.30.mlp.up_proj.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.30.post_attention_layernorm.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.30.self_attn.W_pack.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.30.self_attn.o_proj.weight": "pytorch_model-00014-of-00016.bin",
+    "model.layers.31.input_layernorm.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.31.mlp.down_proj.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.31.mlp.gate_proj.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.31.mlp.up_proj.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.31.post_attention_layernorm.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.31.self_attn.W_pack.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.31.self_attn.o_proj.weight": "pytorch_model-00015-of-00016.bin",
+    "model.layers.4.input_layernorm.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.4.mlp.down_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.4.mlp.gate_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.4.mlp.up_proj.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.4.post_attention_layernorm.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.4.self_attn.W_pack.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.4.self_attn.o_proj.weight": "pytorch_model-00003-of-00016.bin",
+    "model.layers.5.input_layernorm.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.5.mlp.down_proj.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.5.mlp.gate_proj.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.5.mlp.up_proj.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.5.post_attention_layernorm.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.5.self_attn.W_pack.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.5.self_attn.o_proj.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.6.input_layernorm.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.6.mlp.down_proj.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.6.mlp.gate_proj.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.6.mlp.up_proj.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.6.post_attention_layernorm.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.6.self_attn.W_pack.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.6.self_attn.o_proj.weight": "pytorch_model-00004-of-00016.bin",
+    "model.layers.7.input_layernorm.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.7.mlp.down_proj.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.7.mlp.gate_proj.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.7.mlp.up_proj.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.7.post_attention_layernorm.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.7.self_attn.W_pack.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.7.self_attn.o_proj.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.8.input_layernorm.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.8.mlp.down_proj.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.8.mlp.gate_proj.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.8.mlp.up_proj.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.8.post_attention_layernorm.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.8.self_attn.W_pack.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.8.self_attn.o_proj.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.9.input_layernorm.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.9.mlp.down_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.9.mlp.gate_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.9.mlp.up_proj.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.9.post_attention_layernorm.weight": "pytorch_model-00006-of-00016.bin",
+    "model.layers.9.self_attn.W_pack.weight": "pytorch_model-00005-of-00016.bin",
+    "model.layers.9.self_attn.o_proj.weight": "pytorch_model-00005-of-00016.bin",
+    "model.norm.weight": "pytorch_model-00015-of-00016.bin"
+  }
+}