Update README.md

README.md

@@ -16,7 +16,7 @@ More information about Erya dataset can be found here: [RUCAIBox/Erya-dataset ·
 from transformers import BertTokenizer, CPTForConditionalGeneration
 
 tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya")
-model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya")
+model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya4FT")
 
 input_ids = tokenizer("安世字子孺，少以父任为郎。", return_tensors='pt')
 input_ids.pop("token_type_ids")