update model weights

.gitattributes

@@ -43,3 +43,11 @@ pytorch_model-00001-of-00008.bin filter=lfs diff=lfs merge=lfs -text
 pytorch_model-00002-of-00008.bin filter=lfs diff=lfs merge=lfs -text
 pytorch_model-00005-of-00008.bin filter=lfs diff=lfs merge=lfs -text
 pytorch_model.bin.index.json filter=lfs diff=lfs merge=lfs -text
+model-00001-of-00008.safetensors filter=lfs diff=lfs merge=lfs -text
+model-00002-of-00008.safetensors filter=lfs diff=lfs merge=lfs -text
+model-00003-of-00008.safetensors filter=lfs diff=lfs merge=lfs -text
+model-00004-of-00008.safetensors filter=lfs diff=lfs merge=lfs -text
+model-00005-of-00008.safetensors filter=lfs diff=lfs merge=lfs -text
+model-00006-of-00008.safetensors filter=lfs diff=lfs merge=lfs -text
+model-00007-of-00008.safetensors filter=lfs diff=lfs merge=lfs -text
+model-00008-of-00008.safetensors filter=lfs diff=lfs merge=lfs -text

config.json

@@ -2,6 +2,7 @@
   "architectures": [
     "InternLM2ForCausalLM"
   ],
+  "attn_implementation": "eager",
   "auto_map": {
     "AutoConfig": "configuration_internlm2.InternLM2Config",
     "AutoModelForCausalLM": "modeling_internlm2.InternLM2ForCausalLM",
@@ -21,14 +22,11 @@
   "num_key_value_heads": 8,
   "pad_token_id": 2,
   "rms_norm_eps": 1e-05,
-  "rope_scaling": {
-    "factor": 2.0,
-    "type": "dynamic"
-  },
+  "rope_scaling": null,
   "rope_theta": 1000000,
   "tie_word_embeddings": false,
-  "torch_dtype": "float16",
-  "transformers_version": "4.33.2",
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.37.1",
   "use_cache": true,
   "vocab_size": 92544
 }

generation_config.json

@@ -3,5 +3,5 @@
   "bos_token_id": 1,
   "eos_token_id": 2,
   "pad_token_id": 2,
-  "transformers_version": "4.33.2"
+  "transformers_version": "4.37.1"
 }

pytorch_model-00001-of-00008.bin → model-00001-of-00008.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fdd03c1ea040b17ce992eb303d0824f5694932902d0c3fa57b5a8d1df2dc8082
-size 1949342053
+oid sha256:9cec9c25c5f54c545deb16123fc72a0b265fd6bed4c467b4519380a8b215b62b
+size 1949337704

pytorch_model-00002-of-00008.bin → model-00002-of-00008.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:377d66327bcc2dc4b3e5f4e90f24b428c55832ece07194990248647ea65c2db5
-size 1946249825
+oid sha256:4735b4b15a670c33a6aefe70cf39e4ba08468a496167b5233fa43ebcfbd42895
+size 1946242696

pytorch_model-00003-of-00008.bin → model-00003-of-00008.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:98aa01b7ffd7901699b4588b5e6b8f753e66c1874f8b855c36b23de2c95ae9d8
-size 1979786923
+oid sha256:523c7292140bc4c37d5fbd517ec7f9d0aa27b7d22822fdfc730f94f3bf02ebac
+size 1979780440

pytorch_model-00004-of-00008.bin → model-00004-of-00008.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9362abdb2afd077e579a7860ad570769255c2d64af2b0831413578c77a44d779
-size 1946249889
+oid sha256:2dd631ca87a5e516780832e747156e304a3c9bb53f85c4304550bff970d2db26
+size 1946242728

model-00005-of-00008.safetensors

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

model-00006-of-00008.safetensors

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

model-00007-of-00008.safetensors

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

model-00008-of-00008.safetensors

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

model.safetensors.index.json

@@ -0,0 +1,234 @@
+{
+  "metadata": {
+    "total_size": 15475417088
+  },
+  "weight_map": {
+    "model.layers.0.attention.wo.weight": "model-00001-of-00008.safetensors",
+    "model.layers.0.attention.wqkv.weight": "model-00001-of-00008.safetensors",
+    "model.layers.0.attention_norm.weight": "model-00001-of-00008.safetensors",
+    "model.layers.0.feed_forward.w1.weight": "model-00001-of-00008.safetensors",
+    "model.layers.0.feed_forward.w2.weight": "model-00001-of-00008.safetensors",
+    "model.layers.0.feed_forward.w3.weight": "model-00001-of-00008.safetensors",
+    "model.layers.0.ffn_norm.weight": "model-00001-of-00008.safetensors",
+    "model.layers.1.attention.wo.weight": "model-00001-of-00008.safetensors",
+    "model.layers.1.attention.wqkv.weight": "model-00001-of-00008.safetensors",
+    "model.layers.1.attention_norm.weight": "model-00001-of-00008.safetensors",
+    "model.layers.1.feed_forward.w1.weight": "model-00001-of-00008.safetensors",
+    "model.layers.1.feed_forward.w2.weight": "model-00001-of-00008.safetensors",
+    "model.layers.1.feed_forward.w3.weight": "model-00001-of-00008.safetensors",
+    "model.layers.1.ffn_norm.weight": "model-00001-of-00008.safetensors",
+    "model.layers.10.attention.wo.weight": "model-00003-of-00008.safetensors",
+    "model.layers.10.attention.wqkv.weight": "model-00003-of-00008.safetensors",
+    "model.layers.10.attention_norm.weight": "model-00003-of-00008.safetensors",
+    "model.layers.10.feed_forward.w1.weight": "model-00003-of-00008.safetensors",
+    "model.layers.10.feed_forward.w2.weight": "model-00003-of-00008.safetensors",
+    "model.layers.10.feed_forward.w3.weight": "model-00003-of-00008.safetensors",
+    "model.layers.10.ffn_norm.weight": "model-00003-of-00008.safetensors",
+    "model.layers.11.attention.wo.weight": "model-00003-of-00008.safetensors",
+    "model.layers.11.attention.wqkv.weight": "model-00003-of-00008.safetensors",
+    "model.layers.11.attention_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.11.feed_forward.w1.weight": "model-00003-of-00008.safetensors",
+    "model.layers.11.feed_forward.w2.weight": "model-00004-of-00008.safetensors",
+    "model.layers.11.feed_forward.w3.weight": "model-00003-of-00008.safetensors",
+    "model.layers.11.ffn_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.12.attention.wo.weight": "model-00004-of-00008.safetensors",
+    "model.layers.12.attention.wqkv.weight": "model-00004-of-00008.safetensors",
+    "model.layers.12.attention_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.12.feed_forward.w1.weight": "model-00004-of-00008.safetensors",
+    "model.layers.12.feed_forward.w2.weight": "model-00004-of-00008.safetensors",
+    "model.layers.12.feed_forward.w3.weight": "model-00004-of-00008.safetensors",
+    "model.layers.12.ffn_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.13.attention.wo.weight": "model-00004-of-00008.safetensors",
+    "model.layers.13.attention.wqkv.weight": "model-00004-of-00008.safetensors",
+    "model.layers.13.attention_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.13.feed_forward.w1.weight": "model-00004-of-00008.safetensors",
+    "model.layers.13.feed_forward.w2.weight": "model-00004-of-00008.safetensors",
+    "model.layers.13.feed_forward.w3.weight": "model-00004-of-00008.safetensors",
+    "model.layers.13.ffn_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.14.attention.wo.weight": "model-00004-of-00008.safetensors",
+    "model.layers.14.attention.wqkv.weight": "model-00004-of-00008.safetensors",
+    "model.layers.14.attention_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.14.feed_forward.w1.weight": "model-00004-of-00008.safetensors",
+    "model.layers.14.feed_forward.w2.weight": "model-00004-of-00008.safetensors",
+    "model.layers.14.feed_forward.w3.weight": "model-00004-of-00008.safetensors",
+    "model.layers.14.ffn_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.15.attention.wo.weight": "model-00004-of-00008.safetensors",
+    "model.layers.15.attention.wqkv.weight": "model-00004-of-00008.safetensors",
+    "model.layers.15.attention_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.15.feed_forward.w1.weight": "model-00004-of-00008.safetensors",
+    "model.layers.15.feed_forward.w2.weight": "model-00004-of-00008.safetensors",
+    "model.layers.15.feed_forward.w3.weight": "model-00004-of-00008.safetensors",
+    "model.layers.15.ffn_norm.weight": "model-00004-of-00008.safetensors",
+    "model.layers.16.attention.wo.weight": "model-00004-of-00008.safetensors",
+    "model.layers.16.attention.wqkv.weight": "model-00004-of-00008.safetensors",
+    "model.layers.16.attention_norm.weight": "model-00005-of-00008.safetensors",
+    "model.layers.16.feed_forward.w1.weight": "model-00005-of-00008.safetensors",
+    "model.layers.16.feed_forward.w2.weight": "model-00005-of-00008.safetensors",
+    "model.layers.16.feed_forward.w3.weight": "model-00005-of-00008.safetensors",
+    "model.layers.16.ffn_norm.weight": "model-00005-of-00008.safetensors",
+    "model.layers.17.attention.wo.weight": "model-00005-of-00008.safetensors",
+    "model.layers.17.attention.wqkv.weight": "model-00005-of-00008.safetensors",
+    "model.layers.17.attention_norm.weight": "model-00005-of-00008.safetensors",
+    "model.layers.17.feed_forward.w1.weight": "model-00005-of-00008.safetensors",
+    "model.layers.17.feed_forward.w2.weight": "model-00005-of-00008.safetensors",
+    "model.layers.17.feed_forward.w3.weight": "model-00005-of-00008.safetensors",
+    "model.layers.17.ffn_norm.weight": "model-00005-of-00008.safetensors",
+    "model.layers.18.attention.wo.weight": "model-00005-of-00008.safetensors",
+    "model.layers.18.attention.wqkv.weight": "model-00005-of-00008.safetensors",
+    "model.layers.18.attention_norm.weight": "model-00005-of-00008.safetensors",
+    "model.layers.18.feed_forward.w1.weight": "model-00005-of-00008.safetensors",
+    "model.layers.18.feed_forward.w2.weight": "model-00005-of-00008.safetensors",
+    "model.layers.18.feed_forward.w3.weight": "model-00005-of-00008.safetensors",
+    "model.layers.18.ffn_norm.weight": "model-00005-of-00008.safetensors",
+    "model.layers.19.attention.wo.weight": "model-00005-of-00008.safetensors",
+    "model.layers.19.attention.wqkv.weight": "model-00005-of-00008.safetensors",
+    "model.layers.19.attention_norm.weight": "model-00005-of-00008.safetensors",
+    "model.layers.19.feed_forward.w1.weight": "model-00005-of-00008.safetensors",
+    "model.layers.19.feed_forward.w2.weight": "model-00005-of-00008.safetensors",
+    "model.layers.19.feed_forward.w3.weight": "model-00005-of-00008.safetensors",
+    "model.layers.19.ffn_norm.weight": "model-00005-of-00008.safetensors",
+    "model.layers.2.attention.wo.weight": "model-00001-of-00008.safetensors",
+    "model.layers.2.attention.wqkv.weight": "model-00001-of-00008.safetensors",
+    "model.layers.2.attention_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.2.feed_forward.w1.weight": "model-00001-of-00008.safetensors",
+    "model.layers.2.feed_forward.w2.weight": "model-00002-of-00008.safetensors",
+    "model.layers.2.feed_forward.w3.weight": "model-00001-of-00008.safetensors",
+    "model.layers.2.ffn_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.20.attention.wo.weight": "model-00005-of-00008.safetensors",
+    "model.layers.20.attention.wqkv.weight": "model-00005-of-00008.safetensors",
+    "model.layers.20.attention_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.20.feed_forward.w1.weight": "model-00005-of-00008.safetensors",
+    "model.layers.20.feed_forward.w2.weight": "model-00006-of-00008.safetensors",
+    "model.layers.20.feed_forward.w3.weight": "model-00005-of-00008.safetensors",
+    "model.layers.20.ffn_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.21.attention.wo.weight": "model-00006-of-00008.safetensors",
+    "model.layers.21.attention.wqkv.weight": "model-00006-of-00008.safetensors",
+    "model.layers.21.attention_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.21.feed_forward.w1.weight": "model-00006-of-00008.safetensors",
+    "model.layers.21.feed_forward.w2.weight": "model-00006-of-00008.safetensors",
+    "model.layers.21.feed_forward.w3.weight": "model-00006-of-00008.safetensors",
+    "model.layers.21.ffn_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.22.attention.wo.weight": "model-00006-of-00008.safetensors",
+    "model.layers.22.attention.wqkv.weight": "model-00006-of-00008.safetensors",
+    "model.layers.22.attention_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.22.feed_forward.w1.weight": "model-00006-of-00008.safetensors",
+    "model.layers.22.feed_forward.w2.weight": "model-00006-of-00008.safetensors",
+    "model.layers.22.feed_forward.w3.weight": "model-00006-of-00008.safetensors",
+    "model.layers.22.ffn_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.23.attention.wo.weight": "model-00006-of-00008.safetensors",
+    "model.layers.23.attention.wqkv.weight": "model-00006-of-00008.safetensors",
+    "model.layers.23.attention_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.23.feed_forward.w1.weight": "model-00006-of-00008.safetensors",
+    "model.layers.23.feed_forward.w2.weight": "model-00006-of-00008.safetensors",
+    "model.layers.23.feed_forward.w3.weight": "model-00006-of-00008.safetensors",
+    "model.layers.23.ffn_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.24.attention.wo.weight": "model-00006-of-00008.safetensors",
+    "model.layers.24.attention.wqkv.weight": "model-00006-of-00008.safetensors",
+    "model.layers.24.attention_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.24.feed_forward.w1.weight": "model-00006-of-00008.safetensors",
+    "model.layers.24.feed_forward.w2.weight": "model-00006-of-00008.safetensors",
+    "model.layers.24.feed_forward.w3.weight": "model-00006-of-00008.safetensors",
+    "model.layers.24.ffn_norm.weight": "model-00006-of-00008.safetensors",
+    "model.layers.25.attention.wo.weight": "model-00006-of-00008.safetensors",
+    "model.layers.25.attention.wqkv.weight": "model-00006-of-00008.safetensors",
+    "model.layers.25.attention_norm.weight": "model-00007-of-00008.safetensors",
+    "model.layers.25.feed_forward.w1.weight": "model-00007-of-00008.safetensors",
+    "model.layers.25.feed_forward.w2.weight": "model-00007-of-00008.safetensors",
+    "model.layers.25.feed_forward.w3.weight": "model-00007-of-00008.safetensors",
+    "model.layers.25.ffn_norm.weight": "model-00007-of-00008.safetensors",
+    "model.layers.26.attention.wo.weight": "model-00007-of-00008.safetensors",
+    "model.layers.26.attention.wqkv.weight": "model-00007-of-00008.safetensors",
+    "model.layers.26.attention_norm.weight": "model-00007-of-00008.safetensors",
+    "model.layers.26.feed_forward.w1.weight": "model-00007-of-00008.safetensors",
+    "model.layers.26.feed_forward.w2.weight": "model-00007-of-00008.safetensors",
+    "model.layers.26.feed_forward.w3.weight": "model-00007-of-00008.safetensors",
+    "model.layers.26.ffn_norm.weight": "model-00007-of-00008.safetensors",
+    "model.layers.27.attention.wo.weight": "model-00007-of-00008.safetensors",
+    "model.layers.27.attention.wqkv.weight": "model-00007-of-00008.safetensors",
+    "model.layers.27.attention_norm.weight": "model-00007-of-00008.safetensors",
+    "model.layers.27.feed_forward.w1.weight": "model-00007-of-00008.safetensors",
+    "model.layers.27.feed_forward.w2.weight": "model-00007-of-00008.safetensors",
+    "model.layers.27.feed_forward.w3.weight": "model-00007-of-00008.safetensors",
+    "model.layers.27.ffn_norm.weight": "model-00007-of-00008.safetensors",
+    "model.layers.28.attention.wo.weight": "model-00007-of-00008.safetensors",
+    "model.layers.28.attention.wqkv.weight": "model-00007-of-00008.safetensors",
+    "model.layers.28.attention_norm.weight": "model-00007-of-00008.safetensors",
+    "model.layers.28.feed_forward.w1.weight": "model-00007-of-00008.safetensors",
+    "model.layers.28.feed_forward.w2.weight": "model-00007-of-00008.safetensors",
+    "model.layers.28.feed_forward.w3.weight": "model-00007-of-00008.safetensors",
+    "model.layers.28.ffn_norm.weight": "model-00007-of-00008.safetensors",
+    "model.layers.29.attention.wo.weight": "model-00007-of-00008.safetensors",
+    "model.layers.29.attention.wqkv.weight": "model-00007-of-00008.safetensors",
+    "model.layers.29.attention_norm.weight": "model-00008-of-00008.safetensors",
+    "model.layers.29.feed_forward.w1.weight": "model-00007-of-00008.safetensors",
+    "model.layers.29.feed_forward.w2.weight": "model-00008-of-00008.safetensors",
+    "model.layers.29.feed_forward.w3.weight": "model-00007-of-00008.safetensors",
+    "model.layers.29.ffn_norm.weight": "model-00008-of-00008.safetensors",
+    "model.layers.3.attention.wo.weight": "model-00002-of-00008.safetensors",
+    "model.layers.3.attention.wqkv.weight": "model-00002-of-00008.safetensors",
+    "model.layers.3.attention_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.3.feed_forward.w1.weight": "model-00002-of-00008.safetensors",
+    "model.layers.3.feed_forward.w2.weight": "model-00002-of-00008.safetensors",
+    "model.layers.3.feed_forward.w3.weight": "model-00002-of-00008.safetensors",
+    "model.layers.3.ffn_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.30.attention.wo.weight": "model-00008-of-00008.safetensors",
+    "model.layers.30.attention.wqkv.weight": "model-00008-of-00008.safetensors",
+    "model.layers.30.attention_norm.weight": "model-00008-of-00008.safetensors",
+    "model.layers.30.feed_forward.w1.weight": "model-00008-of-00008.safetensors",
+    "model.layers.30.feed_forward.w2.weight": "model-00008-of-00008.safetensors",
+    "model.layers.30.feed_forward.w3.weight": "model-00008-of-00008.safetensors",
+    "model.layers.30.ffn_norm.weight": "model-00008-of-00008.safetensors",
+    "model.layers.31.attention.wo.weight": "model-00008-of-00008.safetensors",
+    "model.layers.31.attention.wqkv.weight": "model-00008-of-00008.safetensors",
+    "model.layers.31.attention_norm.weight": "model-00008-of-00008.safetensors",
+    "model.layers.31.feed_forward.w1.weight": "model-00008-of-00008.safetensors",
+    "model.layers.31.feed_forward.w2.weight": "model-00008-of-00008.safetensors",
+    "model.layers.31.feed_forward.w3.weight": "model-00008-of-00008.safetensors",
+    "model.layers.31.ffn_norm.weight": "model-00008-of-00008.safetensors",
+    "model.layers.4.attention.wo.weight": "model-00002-of-00008.safetensors",
+    "model.layers.4.attention.wqkv.weight": "model-00002-of-00008.safetensors",
+    "model.layers.4.attention_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.4.feed_forward.w1.weight": "model-00002-of-00008.safetensors",
+    "model.layers.4.feed_forward.w2.weight": "model-00002-of-00008.safetensors",
+    "model.layers.4.feed_forward.w3.weight": "model-00002-of-00008.safetensors",
+    "model.layers.4.ffn_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.5.attention.wo.weight": "model-00002-of-00008.safetensors",
+    "model.layers.5.attention.wqkv.weight": "model-00002-of-00008.safetensors",
+    "model.layers.5.attention_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.5.feed_forward.w1.weight": "model-00002-of-00008.safetensors",
+    "model.layers.5.feed_forward.w2.weight": "model-00002-of-00008.safetensors",
+    "model.layers.5.feed_forward.w3.weight": "model-00002-of-00008.safetensors",
+    "model.layers.5.ffn_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.6.attention.wo.weight": "model-00002-of-00008.safetensors",
+    "model.layers.6.attention.wqkv.weight": "model-00002-of-00008.safetensors",
+    "model.layers.6.attention_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.6.feed_forward.w1.weight": "model-00002-of-00008.safetensors",
+    "model.layers.6.feed_forward.w2.weight": "model-00002-of-00008.safetensors",
+    "model.layers.6.feed_forward.w3.weight": "model-00002-of-00008.safetensors",
+    "model.layers.6.ffn_norm.weight": "model-00002-of-00008.safetensors",
+    "model.layers.7.attention.wo.weight": "model-00002-of-00008.safetensors",
+    "model.layers.7.attention.wqkv.weight": "model-00002-of-00008.safetensors",
+    "model.layers.7.attention_norm.weight": "model-00003-of-00008.safetensors",
+    "model.layers.7.feed_forward.w1.weight": "model-00003-of-00008.safetensors",
+    "model.layers.7.feed_forward.w2.weight": "model-00003-of-00008.safetensors",
+    "model.layers.7.feed_forward.w3.weight": "model-00003-of-00008.safetensors",
+    "model.layers.7.ffn_norm.weight": "model-00003-of-00008.safetensors",
+    "model.layers.8.attention.wo.weight": "model-00003-of-00008.safetensors",
+    "model.layers.8.attention.wqkv.weight": "model-00003-of-00008.safetensors",
+    "model.layers.8.attention_norm.weight": "model-00003-of-00008.safetensors",
+    "model.layers.8.feed_forward.w1.weight": "model-00003-of-00008.safetensors",
+    "model.layers.8.feed_forward.w2.weight": "model-00003-of-00008.safetensors",
+    "model.layers.8.feed_forward.w3.weight": "model-00003-of-00008.safetensors",
+    "model.layers.8.ffn_norm.weight": "model-00003-of-00008.safetensors",
+    "model.layers.9.attention.wo.weight": "model-00003-of-00008.safetensors",
+    "model.layers.9.attention.wqkv.weight": "model-00003-of-00008.safetensors",
+    "model.layers.9.attention_norm.weight": "model-00003-of-00008.safetensors",
+    "model.layers.9.feed_forward.w1.weight": "model-00003-of-00008.safetensors",
+    "model.layers.9.feed_forward.w2.weight": "model-00003-of-00008.safetensors",
+    "model.layers.9.feed_forward.w3.weight": "model-00003-of-00008.safetensors",
+    "model.layers.9.ffn_norm.weight": "model-00003-of-00008.safetensors",
+    "model.norm.weight": "model-00008-of-00008.safetensors",
+    "model.tok_embeddings.weight": "model-00001-of-00008.safetensors",
+    "output.weight": "model-00008-of-00008.safetensors"
+  }
+}

pytorch_model-00005-of-00008.bin

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

pytorch_model-00006-of-00008.bin

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

pytorch_model-00007-of-00008.bin

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

pytorch_model-00008-of-00008.bin

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

pytorch_model.bin.index.json

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

special_tokens_map.json

@@ -7,8 +7,32 @@
     "<|interpreter|>",
     "<|plugin|>"
   ],
-  "bos_token": "<s>",
-  "eos_token": "</s>",
-  "pad_token": "</s>",
-  "unk_token": "<unk>"
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
 }

tokenization_internlm2_fast.py

@@ -56,14 +56,14 @@ class InternLM2Converter(SpmConverter):
         return unk_id
 
     def decoder(self, replacement, add_prefix_space):
-        decoders_sequence = [
-            decoders.Replace("▁", " "),
-            decoders.ByteFallback(),
-            decoders.Fuse(),
-        ]
-        if self.proto.normalizer_spec.add_dummy_prefix:
-            decoders_sequence.append(decoders.Strip(content=" ", left=1))
-        return decoders.Sequence(decoders_sequence)
+        return decoders.Sequence(
+            [
+                decoders.Replace("▁", " "),
+                decoders.ByteFallback(),
+                decoders.Fuse(),
+                decoders.Strip(content=" ", left=1),
+            ]
+        )
 
     def tokenizer(self, proto):
         model_type = proto.trainer_spec.model_type

tokenizer_config.json

@@ -1,17 +1,6 @@
 {
-  "auto_map": {
-    "AutoTokenizer": [
-      "tokenization_internlm2.InternLM2Tokenizer",
-      "tokenization_internlm2_fast.InternLM2TokenizerFast"
-    ]
-  },
-  "bos_token": "<s>",
-  "clean_up_tokenization_spaces": false,
-  "eos_token": "</s>",
-  "model_max_length": 1000000000000000019884624838656,
-  "pad_token": "</s>",
-  "tokenizer_class": "InternLM2Tokenizer",
-  "unk_token": "<unk>",
+  "add_bos_token": true,
+  "add_eos_token": false,
   "added_tokens_decoder": {
     "0": {
       "content": "<unk>",
@@ -37,48 +26,48 @@
       "single_word": false,
       "special": true
     },
-    "92543": {
-      "content": "<|im_start|>",
+    "92538": {
+      "content": "<|plugin|>",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     },
-    "92542": {
-      "content": "<|im_end|>",
+    "92539": {
+      "content": "<|interpreter|>",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     },
-    "92541": {
-      "content": "<|action_start|>",
+    "92540": {
+      "content": "<|action_end|>",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     },
-    "92540": {
-      "content": "<|action_end|>",
+    "92541": {
+      "content": "<|action_start|>",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     },
-    "92539": {
-      "content": "<|interpreter|>",
+    "92542": {
+      "content": "<|im_end|>",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
       "single_word": false,
       "special": true
     },
-    "92538": {
-      "content": "<|plugin|>",
+    "92543": {
+      "content": "<|im_start|>",
       "lstrip": false,
       "normalized": false,
       "rstrip": false,
@@ -94,5 +83,20 @@
     "<|interpreter|>",
     "<|plugin|>"
   ],
-  "chat_template": "{{ bos_token }}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}"
-}
+  "auto_map": {
+    "AutoTokenizer": [
+      "tokenization_internlm2.InternLM2Tokenizer",
+      "tokenization_internlm2_fast.InternLM2TokenizerFast"
+    ]
+  },
+  "bos_token": "<s>",
+  "chat_template": "{{ bos_token }}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
+  "clean_up_tokenization_spaces": false,
+  "decode_with_prefix_space": false,
+  "eos_token": "</s>",
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "</s>",
+  "sp_model_kwargs": null,
+  "tokenizer_class": "InternLM2Tokenizer",
+  "unk_token": "<unk>"
+}