macadeliccc commited on
Commit
beb273a
1 Parent(s): e971d65

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -42,8 +42,8 @@ import torch
42
  from llm2vec import LLM2Vec
43
 
44
  l2v = LLM2Vec.from_pretrained(
45
- "McGill-NLP/LLM2Vec-Mistral-7B-Instruct-v2-mntp",
46
- peft_model_name_or_path="McGill-NLP/LLM2Vec-Mistral-7B-Instruct-v2-mntp-unsup-simcse",
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
  )