rightyonghu commited on
Commit
4e440ba
•
1 Parent(s): 5be18a0

update model

Browse files
Files changed (3) hide show
  1. README.md +2 -9
  2. config.json +2 -2
  3. pytorch_model.bin +2 -2
README.md CHANGED
@@ -18,18 +18,11 @@ a series of experiments have been conducted to check the accuracy of the convers
18
  - Pytorch Conversion repo: https://github.com/nghuyong/ERNIE-Pytorch
19
 
20
  ## How to use
21
- 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)
22
- **OR** you can re-install the transformers from my changed branch.
23
- ```bash
24
- pip uninstall transformers # optional
25
- pip install git+https://github.com/nghuyong/transformers@add_task_type_id # reinstall
26
- ```
27
- Then you can load ERNIE-3.0 model as before:
28
  ```Python
29
- from transformers import BertTokenizer, BertModel
30
 
31
  tokenizer = BertTokenizer.from_pretrained("nghuyong/ernie-3.0-medium-zh")
32
- model = BertModel.from_pretrained("nghuyong/ernie-3.0-medium-zh")
33
  ```
34
 
35
  ## Citation
 
18
  - Pytorch Conversion repo: https://github.com/nghuyong/ERNIE-Pytorch
19
 
20
  ## How to use
 
 
 
 
 
 
 
21
  ```Python
22
+ from transformers import BertTokenizer, ErnieModel
23
 
24
  tokenizer = BertTokenizer.from_pretrained("nghuyong/ernie-3.0-medium-zh")
25
+ model = ErnieModel.from_pretrained("nghuyong/ernie-3.0-medium-zh")
26
  ```
27
 
28
  ## Citation
config.json CHANGED
@@ -13,8 +13,8 @@
13
  "use_task_id": true,
14
  "vocab_size": 40000,
15
  "layer_norm_eps": 1e-05,
16
- "model_type": "bert",
17
  "architectures": [
18
- "BertModel"
19
  ]
20
  }
 
13
  "use_task_id": true,
14
  "vocab_size": 40000,
15
  "layer_norm_eps": 1e-05,
16
+ "model_type": "ernie",
17
  "architectures": [
18
+ "ErnieForMaskedLM"
19
  ]
20
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8feaf40dd4338f35c78c06ad8abed5aaf0452acc8f3d0f509c089e2b9ec63090
3
- size 299379829
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65f35e2ada39e50eea624f2e85c9aa95a87a8189cf5f2562c485e09243298355
3
+ size 299379957