Update README.md
Browse files
README.md
CHANGED
@@ -39,10 +39,9 @@ import numpy as np
|
|
39 |
|
40 |
sentence = '清华大学位于'
|
41 |
max_length = 32
|
42 |
-
model_pretrained_weight_path = '/home/' # 预训练模型权重路径
|
43 |
|
44 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
45 |
-
model = RoFormerModel.from_pretrained(
|
46 |
|
47 |
for i in range(max_length):
|
48 |
encode = torch.tensor(
|
|
|
39 |
|
40 |
sentence = '清华大学位于'
|
41 |
max_length = 32
|
|
|
42 |
|
43 |
+
tokenizer = AutoTokenizer.from_pretrained("IDEA-CCNL/Zhouwenwang-110M")
|
44 |
+
model = RoFormerModel.from_pretrained("IDEA-CCNL/Zhouwenwang-110M")
|
45 |
|
46 |
for i in range(max_length):
|
47 |
encode = torch.tensor(
|