Update config.json

config.json

@@ -1,8 +1,14 @@
 {
-  "_name_or_path": "sgpt2_m7_sq2048_e5ds_bbcc_bs2048_gc",
+  "_name_or_path": "GritLM/GritLM-7B",
+  "model_type": "gritlm",
   "architectures": [
     "MistralForCausalLM"
   ],
+  "auto_map": {
+    "AutoModel": "modeling_gritlm7b.MistralModel",
+    "AutoModelForCausalLM": "modeling_gritlm7b.MistralForCausalLM",
+    "AutoModelForSequenceClassification": "modeling_gritlm7b.MistralForSequenceClassification"
+  },
   "attention_dropout": 0.0,
   "bos_token_id": 1,
   "eos_token_id": 2,
@@ -17,7 +23,6 @@
     "LABEL_0": 0
   },
   "max_position_embeddings": 32768,
-  "model_type": "mistral",
   "num_attention_heads": 32,
   "num_hidden_layers": 32,
   "num_key_value_heads": 8,