nvidia
/

llama-nemotron-embed-1b-v2

Feature Extraction

text-embeddings

semantic-search

Model card Files Files and versions

Mengyao00 commited on 30 days ago

Commit

92f18ce

·

verified ·

1 Parent(s): 684c706

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -77,7 +77,7 @@ def average_pool(last_hidden_states, attention_mask):
 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 = 'query: '

 tokenizer = AutoTokenizer.from_pretrained("nvidia/llama-3.2-nv-embedqa-1b-v2")
+model = AutoModel.from_pretrained("nvidia/llama-3.2-nv-embedqa-1b-v2", trust_remote_code=True)
 model = model.to("cuda:0")
 model.eval()
 query_prefix = 'query: '