Update README.md

README.md

@@ -15,8 +15,8 @@ so `trust_remote_code=True` is required to load model implementation in this rep
 ```python
 from transformers import TextGenerationPipeline, AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("imone/pangu_2_6B", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("imone/pangu_2_6B", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("ModelCloud/pangu_alpha_2_6B", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("ModelCloud/pangu_alpha_2_6B", trust_remote_code=True)
 
 text_generator = TextGenerationPipeline(model, tokenizer)
 # greedy search