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-micro-zh")
-model = BertModel.from_pretrained("nghuyong/ernie-3.0-micro-zh")
+model = ErnieModel.from_pretrained("nghuyong/ernie-3.0-micro-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"
+        "ErnieModel"
     ]
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:36a4ae1eb13f0c51cf0d83c0972063dc6dbe7c1ab33c19a7bfa1ba4b540b9716
-size 93032531
+oid sha256:12c6078f4bf79e46a3c02d1c3784077e206261818d579bb5965793bcd94f9aae
+size 93032595