update readme

README.md

@@ -98,14 +98,14 @@ For our pretrained model (Moonlight-16B-A3B):
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_path = "path-to-your-checkpoint"
+model_name = "moonshotai/Moonlight-16B-A3B"
 model = AutoModelForCausalLM.from_pretrained(
-    model_path,
+    model_name,
     torch_dtype="auto",
     device_map="auto",
     trust_remote_code=True,
 )
-tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
 prompt = "1+1=2, 1+2="
 inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True).to(model.device)
@@ -118,14 +118,14 @@ For our instruct model (Moonlight-16B-A3B-Instruct):
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_path = "path-to-your-checkpoint"
+model_name = "moonshotai/Moonlight-16B-A3B-Instruct"
 model = AutoModelForCausalLM.from_pretrained(
-    model_path,
+    model_name,
     torch_dtype="auto",
     device_map="auto",
     trust_remote_code=True
 )
-tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
 prompt = "Give me a short introduction to large language model."
 messages = [