Update README.md

README.md

@@ -33,14 +33,14 @@ Now, let's start to download the model.
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_path = "Mike0307/Phi-3-mini-4k-instruct-chinese-lora"
-base_model = AutoModelForCausalLM.from_pretrained(
-    model_path, 
+model_id = "Mike0307/Phi-3-mini-4k-instruct-chinese-lora"
+model = AutoModelForCausalLM.from_pretrained(
+    model_id, 
     device_map="mps", # FIX mps if not MacOS
     torch_dtype=torch.float32, 
     trust_remote_code=True,
 )
-tokenizer = AutoTokenizer.from_pretrained(model_path)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
 ```
 
 ## Example of inference
@@ -49,7 +49,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
 input_text = "<|user|>將這五種動物分成兩組。\n老虎、鯊魚、大象、鯨魚、袋鼠 <|end|>\n<|assistant|>"
 inputs = tokenizer(input_text, return_tensors="pt").to(torch.device("mps")) # FIX mps if not MacOS
 
-outputs = base_model.generate(
+outputs = model.generate(
     **inputs, 
     temperature = 0.0,
     max_length = 500,