Transformers
PyTorch
Chinese
megatron-bert
suolyer commited on
Commit
d6f9d68
·
1 Parent(s): 6c4d83b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
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(model_pretrained_weight_path)
45
- model = RoFormerModel.from_pretrained(model_pretrained_weight_path)
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(