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-nano-zh")
-model = BertModel.from_pretrained("nghuyong/ernie-3.0-nano-zh")
+model = ErnieModel.from_pretrained("nghuyong/ernie-3.0-nano-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:46b9acbe991cb5f4f433a217be2fd5506aefaf6908793c3918ba31e10824d0cb
-size 71279059
+oid sha256:513bc520119d9d9f7fbecced215b618691e4dff3ec65b83658024f5018cacd50
+size 71279123