MILVLG
/

Imp-v1.5-4B-Phi3

Text Generation

Model card Files Files and versions Community

Oyoy1235 commited on Jun 21

Commit

2d9d540

•

1 Parent(s): 67394b9

update readme

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ torch.set_default_device("cuda")
 #Create model
 model = AutoModelForCausalLM.from_pretrained(
-    "MILVLG/Imp-v1.5-4B-Phi3",
     torch_dtype=torch.float16,
     device_map="auto",
     trust_remote_code=True)

 #Create model
 model = AutoModelForCausalLM.from_pretrained(
+    "MILVLG/Imp-v1.5-4B-Phi3/",
     torch_dtype=torch.float16,
     device_map="auto",
     trust_remote_code=True)