Update README.md

README.md

@@ -28,8 +28,8 @@ You can easily call the model with the following code:
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-tokenizer = AutoTokenizer.from_pretrained("Chuxin/Chuxin-1.6B-1M", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("Chuxin/Chuxin-1.6B-1M", device_map="auto", trust_remote_code=True, bf16=True).eval()
+tokenizer = AutoTokenizer.from_pretrained("chuxin-llm/Chuxin-1.6B-1M", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("chuxin-llm/Chuxin-1.6B-1M", device_map="auto", trust_remote_code=True, bf16=True).eval()
 inputs = tokenizer('蒙古国的首都是乌兰巴托（Ulaanbaatar）\n冰岛的首都是雷克雅未克（Reykjavik）\n埃塞俄比亚的首都是', return_tensors='pt')
 inputs = inputs.to(model.device)
 pred = model.generate(**inputs, max_new_tokens=20, do_sample=False)