straybird commited on
Commit
35350d4
·
1 Parent(s): 0cf81d7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -16,7 +16,7 @@ More information about Erya dataset can be found here: [RUCAIBox/Erya-dataset ·
16
  from transformers import BertTokenizer, CPTForConditionalGeneration
17
 
18
  tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya")
19
- model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya4FT")
20
 
21
  input_ids = tokenizer("安世字子孺,少以父任为郎。", return_tensors='pt')
22
  input_ids.pop("token_type_ids")
 
16
  from transformers import BertTokenizer, CPTForConditionalGeneration
17
 
18
  tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya")
19
+ model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya")
20
 
21
  input_ids = tokenizer("安世字子孺,少以父任为郎。", return_tensors='pt')
22
  input_ids.pop("token_type_ids")