junnyu
/

roformer_chinese_sim_char_small

Text Generation

Model card Files Files and versions

junnyu commited on Apr 2, 2022

Commit

0702a52

·

1 Parent(s): 05b8ee5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -18,7 +18,7 @@ from transformers import BertTokenizer
 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 pretrained_model = "junnyu/roformer_chinese_sim_char_base"
-tokenizer = RoFormerTokenizer.from_pretrained(pretrained_model)
 config = RoFormerConfig.from_pretrained(pretrained_model)
 config.is_decoder = True
 config.eos_token_id = tokenizer.sep_token_id

 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 pretrained_model = "junnyu/roformer_chinese_sim_char_base"
+tokenizer = BertTokenizer.from_pretrained(pretrained_model)
 config = RoFormerConfig.from_pretrained(pretrained_model)
 config.is_decoder = True
 config.eos_token_id = tokenizer.sep_token_id