junnyu commited on
Commit
51fa942
1 Parent(s): 1f6288a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -3
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 tokenization_wobert import WoBertTokenizer
 
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},