Update README.md

README.md

@@ -21,7 +21,7 @@ As we all know, the single direction language model based on decoder structure h
 ### load model
 ```python 
 from transformers import GPT2Tokenizer, GPT2Model
-tokenizer = GPT2Tokenizer.from_pretrained('IDEA-CCNL/Wenzhong-GPT2-3.5B')
+tokenizer = GPT2Tokenizer.from_pretrained('IDEA-CCNL/Wenzhong2.0-GPT2-3.5B-chinese')
 model = GPT2Model.from_pretrained('IDEA-CCNL/Wenzhong-GPT2-3.5B')
 text = "Replace me by any text you'd like."
 encoded_input = tokenizer(text, return_tensors='pt')
@@ -31,7 +31,7 @@ output = model(**encoded_input)
 ```python
 from transformers import pipeline, set_seed
 set_seed(55)
-generator = pipeline('text-generation', model='IDEA-CCNL/Wenzhong-GPT2-3.5B')
+generator = pipeline('text-generation', model='IDEA-CCNL/Wenzhong2.0-GPT2-3.5B-chinese')
 generator("北京位于", max_length=30, num_return_sequences=1)
 
 ```