Update config.json

config.json

@@ -1,17 +1,11 @@
 {
-  "_name_or_path": "/data/.model_and_tokenizer_cache/eea825a7e388c292874bd28736f8bcd70ec7526432cd66fa94c540bd27834d22",
   "architectures": [
-    "NewModel"
+    "GteModel"
   ],
   "attention_probs_dropout_prob": 0.0,
   "auto_map": {
-    "AutoConfig": "Alibaba-NLP/new-impl--configuration.NewConfig",
-    "AutoModel": "Alibaba-NLP/new-impl--modeling.NewModel",
-    "AutoModelForMaskedLM": "Alibaba-NLP/new-impl--modeling.NewForMaskedLM",
-    "AutoModelForMultipleChoice": "Alibaba-NLP/new-impl--modeling.NewForMultipleChoice",
-    "AutoModelForQuestionAnswering": "Alibaba-NLP/new-impl--modeling.NewForQuestionAnswering",
-    "AutoModelForSequenceClassification": "Alibaba-NLP/new-impl--modeling.NewForSequenceClassification",
-    "AutoModelForTokenClassification": "Alibaba-NLP/new-impl--modeling.NewForTokenClassification"
+    "AutoConfig": "configuration_hf_alibaba_nlp_gte.GteConfig",
+    "AutoModel": "modeling_hf_alibaba_nlp_gte.GteModel"
   },
   "classifier_dropout": 0.1,
   "hidden_act": "gelu",
@@ -24,7 +18,7 @@
   "logn_attention_clip1": false,
   "logn_attention_scale": false,
   "max_position_embeddings": 8192,
-  "model_type": "new",
+  "model_type": "gte",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pack_qkv": true,
@@ -38,4 +32,4 @@
   "unpad_inputs": "true",
   "use_memory_efficient_attention": "true",
   "vocab_size": 250048
-}
+}