Update config.json

config.json

@@ -1,11 +1,11 @@
 {
   "architectures": [
-    "InternLM2ForRewardModel"
+    "InternLM2ForCausalLM"
   ],
-  "attn_implementation": "eager",
   "auto_map": {
     "AutoConfig": "configuration_internlm2.InternLM2Config",
-    "AutoModel": "modeling_internlm2.InternLM2ForRewardModel"
+    "AutoModelForCausalLM": "modeling_internlm2.InternLM2ForCausalLM",
+    "AutoModel": "modeling_internlm2.InternLM2ForSequenceClassification"
   },
   "bias": false,
   "bos_token_id": 1,
@@ -20,13 +20,13 @@
   "num_hidden_layers": 32,
   "num_key_value_heads": 8,
   "pad_token_id": 2,
-  "reward_token_id": 92527,
   "rms_norm_eps": 1e-05,
-  "rope_scaling": null,
   "rope_theta": 1000000,
+  "rope_scaling": null,
   "tie_word_embeddings": false,
-  "torch_dtype": "float16",
-  "transformers_version": "4.41.2",
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.41.0",
   "use_cache": true,
-  "vocab_size": 92544
-}
+  "vocab_size": 92544,
+  "pretraining_tp": 1
+}