Update README.md

README.md

@@ -18,7 +18,7 @@ The base-model [microsoft/Phi-3-mini-4k-instruct](https://huggingface.co/microso
 the latest dev-version transformers and torch.<br>
 Also, it needs *trust_remote_code=True* as an argument of the from_pretrained() function.
 ```
-pip install git+https://github.com/huggingface/transformers
+pip install git+https://github.com/huggingface/transformers accelerate
 pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
 ```
 
@@ -37,7 +37,7 @@ model_id = "Mike0307/Phi-3-mini-4k-instruct-chinese-lora"
 model = AutoModelForCausalLM.from_pretrained(
     model_id, 
     device_map="mps", # Change mps if not MacOS
-    torch_dtype=torch.float32, 
+    torch_dtype=torch.float32,  # try float16 for M1 chip
     trust_remote_code=True,
 )
 tokenizer = AutoTokenizer.from_pretrained(model_id)