update model

README.md

@@ -19,7 +19,7 @@ More detail: https://arxiv.org/abs/2010.12148
 This released Pytorch model is converted from the officially released PaddlePaddle ERNIE model and 
 a series of experiments have been conducted to check the accuracy of the conversion.
 
-- Official PaddlePaddle ERNIE repo: https://github.com/PaddlePaddle/ERNIE
+- Official PaddlePaddle ERNIE repo: https://github.com/PaddlePaddle/PaddleNLP/blob/develop/paddlenlp/transformers/ernie_gram/modeling.py
 - Pytorch Conversion repo:  https://github.com/nghuyong/ERNIE-Pytorch
 
 ## How to use

config.json

@@ -1,29 +1,19 @@
 {
-  "_name_or_path": "./convert",
-  "architectures": [
-    "BertModel"
-  ],
-  "attention_probs_dropout_prob": 0.1,
-  "classifier_dropout": null,
-  "emb_size": 768,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.1,
-  "hidden_size": 768,
-  "initializer_range": 0.02,
-  "intermediate_size": 3072,
-  "layer_norm_eps": 1e-05,
-  "max_position_embeddings": 512,
-  "model_type": "bert",
-  "num_attention_heads": 12,
-  "num_hidden_layers": 12,
-  "out_emb_size": 128,
-  "pad_token_id": 0,
-  "position_embedding_type": "absolute",
-  "sent_type_vocab_size": 2,
-  "task_type_vocab_size": 3,
-  "torch_dtype": "float32",
-  "transformers_version": "4.17.0",
-  "type_vocab_size": 2,
-  "use_cache": true,
-  "vocab_size": 18018
-}
+    "attention_probs_dropout_prob": 0.1,
+    "emb_size": 768,
+    "hidden_act": "gelu",
+    "hidden_dropout_prob": 0.1,
+    "hidden_size": 768,
+    "initializer_range": 0.02,
+    "max_position_embeddings": 512,
+    "num_attention_heads": 12,
+    "num_hidden_layers": 12,
+    "type_vocab_size": 2,
+    "vocab_size": 18018,
+    "layer_norm_eps": 1e-05,
+    "model_type": "bert",
+    "architectures": [
+        "BertModel"
+    ],
+    "intermediate_size": 3072
+}

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:45aa8ada50dd35e5ff0b92c1ffe3e1c63f56f96d4248480222107871204906eb
-size 399596081
+oid sha256:dd65658f16c9f5b768353a7b83e643103b6cf4ca125cf57cbc998679be846931
+size 399579827