MILVLG
/

Imp-v1.5-3B-Phi2

Text Generation

Model card Files Files and versions Community

Oyoy1235 commited on Jun 21

Commit

cfa11ef

•

1 Parent(s): 1496057

update readme

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

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

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