Update README.md
Browse files
README.md
CHANGED
@@ -9,13 +9,18 @@ inference: false
|
|
9 |
https://github.com/ZhuiyiTechnology/WoBERT
|
10 |
### pytorch版本
|
11 |
https://github.com/JunnYu/WoBERT_pytorch
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
## 使用
|
13 |
```python
|
14 |
-
git clone https://github.com/JunnYu/WoBERT_pytorch
|
15 |
-
cd WoBERT_pytorch
|
16 |
import torch
|
17 |
from transformers import BertForMaskedLM as WoBertForMaskedLM
|
18 |
-
from
|
|
|
19 |
pretrained_model_or_path_list = [
|
20 |
"junnyu/wobert_chinese_plus_base", "junnyu/wobert_chinese_base"
|
21 |
]
|
@@ -41,7 +46,9 @@ for path in pretrained_model_or_path_list:
|
|
41 |
# WoBERT 今天[天气||阳光||天||心情||空气]很好,我[想||要||就||准备||也]去公园玩。
|
42 |
```
|
43 |
## 引用
|
|
|
44 |
Bibtex:
|
|
|
45 |
```tex
|
46 |
@techreport{zhuiyiwobert,
|
47 |
title={WoBERT: Word-based Chinese BERT model - ZhuiyiAI},
|
|
|
9 |
https://github.com/ZhuiyiTechnology/WoBERT
|
10 |
### pytorch版本
|
11 |
https://github.com/JunnYu/WoBERT_pytorch
|
12 |
+
|
13 |
+
## 安装(主要为了安装WoBertTokenizer)
|
14 |
+
```bash
|
15 |
+
pip install git+https://github.com/JunnYu/RoFormer_pytorch.git
|
16 |
+
```
|
17 |
+
|
18 |
## 使用
|
19 |
```python
|
|
|
|
|
20 |
import torch
|
21 |
from transformers import BertForMaskedLM as WoBertForMaskedLM
|
22 |
+
from roformer import WoBertTokenizer
|
23 |
+
|
24 |
pretrained_model_or_path_list = [
|
25 |
"junnyu/wobert_chinese_plus_base", "junnyu/wobert_chinese_base"
|
26 |
]
|
|
|
46 |
# WoBERT 今天[天气||阳光||天||心情||空气]很好,我[想||要||就||准备||也]去公园玩。
|
47 |
```
|
48 |
## 引用
|
49 |
+
|
50 |
Bibtex:
|
51 |
+
|
52 |
```tex
|
53 |
@techreport{zhuiyiwobert,
|
54 |
title={WoBERT: Word-based Chinese BERT model - ZhuiyiAI},
|