Update README.md
Browse files
README.md
CHANGED
|
@@ -6,7 +6,6 @@ tags:
|
|
| 6 |
- tf2.0
|
| 7 |
inference: False
|
| 8 |
---
|
| 9 |
-
|
| 10 |
# 下载这个roformer代码
|
| 11 |
- https://github.com/JunnYu/RoFormer_pytorch/files/8402725/roformer.zip
|
| 12 |
|
|
@@ -19,7 +18,7 @@ from transformers import BertTokenizer
|
|
| 19 |
|
| 20 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 21 |
pretrained_model = "junnyu/roformer_chinese_sim_char_base"
|
| 22 |
-
tokenizer =
|
| 23 |
config = RoFormerConfig.from_pretrained(pretrained_model)
|
| 24 |
config.is_decoder = True
|
| 25 |
config.eos_token_id = tokenizer.sep_token_id
|
|
|
|
| 6 |
- tf2.0
|
| 7 |
inference: False
|
| 8 |
---
|
|
|
|
| 9 |
# 下载这个roformer代码
|
| 10 |
- https://github.com/JunnYu/RoFormer_pytorch/files/8402725/roformer.zip
|
| 11 |
|
|
|
|
| 18 |
|
| 19 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 20 |
pretrained_model = "junnyu/roformer_chinese_sim_char_base"
|
| 21 |
+
tokenizer = BertTokenizer.from_pretrained(pretrained_model)
|
| 22 |
config = RoFormerConfig.from_pretrained(pretrained_model)
|
| 23 |
config.is_decoder = True
|
| 24 |
config.eos_token_id = tokenizer.sep_token_id
|