Mike0307
/

Phi-3-mini-4k-instruct-chinese-lora

Text Generation

Inference Endpoints

Model card Files Files and versions Community

Mike0307 commited on May 17

Commit

cd25aff

•

1 Parent(s): 735cb24

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 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,
 )

 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,
     trust_remote_code=True,
 )