macadeliccc
commited on
Commit
•
beb273a
1
Parent(s):
e971d65
Update README.md
Browse files
README.md
CHANGED
@@ -42,8 +42,8 @@ import torch
|
|
42 |
from llm2vec import LLM2Vec
|
43 |
|
44 |
l2v = LLM2Vec.from_pretrained(
|
45 |
-
"
|
46 |
-
peft_model_name_or_path="
|
47 |
device_map="cuda" if torch.cuda.is_available() else "cpu",
|
48 |
torch_dtype=torch.bfloat16,
|
49 |
)
|
|
|
42 |
from llm2vec import LLM2Vec
|
43 |
|
44 |
l2v = LLM2Vec.from_pretrained(
|
45 |
+
"macadeliccc/dolphin-2.9-llama3-8b-emb",
|
46 |
+
peft_model_name_or_path="macadeliccc/dolphin-2.9-llama3-8b-emb",
|
47 |
device_map="cuda" if torch.cuda.is_available() else "cpu",
|
48 |
torch_dtype=torch.bfloat16,
|
49 |
)
|