update model

README.md

@@ -18,18 +18,11 @@ a series of experiments have been conducted to check the accuracy of the convers
 - Pytorch Conversion repo:  https://github.com/nghuyong/ERNIE-Pytorch
 
 ## How to use
-If you want to use ernie-3.0 series models, you need to add `task_type_id` to BERT model following this [MR](https://github.com/huggingface/transformers/pull/18686/files) 
-**OR** you can re-install the transformers from my changed branch.
-```bash
-pip uninstall transformers # optional
-pip install git+https://github.com/nghuyong/transformers@add_task_type_id # reinstall
-```
-Then you can load ERNIE-3.0 model as before:
 ```Python
-from transformers import BertTokenizer, BertModel
+from transformers import BertTokenizer, ErnieModel
 
 tokenizer = BertTokenizer.from_pretrained("nghuyong/ernie-3.0-medium-zh")
-model = BertModel.from_pretrained("nghuyong/ernie-3.0-medium-zh")
+model = ErnieModel.from_pretrained("nghuyong/ernie-3.0-medium-zh")
 ```
 
 ## Citation

config.json

@@ -13,8 +13,8 @@
     "use_task_id": true,
     "vocab_size": 40000,
     "layer_norm_eps": 1e-05,
-    "model_type": "bert",
+    "model_type": "ernie",
     "architectures": [
-        "BertModel"
+        "ErnieForMaskedLM"
     ]
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8feaf40dd4338f35c78c06ad8abed5aaf0452acc8f3d0f509c089e2b9ec63090
-size 299379829
+oid sha256:65f35e2ada39e50eea624f2e85c9aa95a87a8189cf5f2562c485e09243298355
+size 299379957