nvidia
/

llama-nemotron-embed-1b-v2

Feature Extraction

text-embeddings

semantic-search

Model card Files Files and versions

Mengyao00 commited on 17 days ago

Commit

b4cce60

·

verified ·

1 Parent(s): b451b51

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -80,7 +80,7 @@ tokenizer = AutoTokenizer.from_pretrained("nvidia/llama-3.2-nv-embedqa-1b-v2")
 model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True)
 model = model.to("cuda:0")
 model.eval()
-query_prefix = 'Given a web search query, retrieve relevant passages that answer the query: '
 document_prefix = 'passage: '

 model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True)
 model = model.to("cuda:0")
 model.eval()
+query_prefix = 'query: '
 document_prefix = 'passage: '