nvidia
/

llama-embed-nemotron-8b

Feature Extraction

sentence-transformers

sentence-similarity

text-embeddings-inference

Model card Files Files and versions

baseten-admin commited on Jan 6

Commit

b81e55d

·

verified ·

1 Parent(s): 5129df8

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -36,5 +36,6 @@
   "torch_dtype": "bfloat16",
   "transformers_version": "4.51.0",
   "use_cache": false,
-  "vocab_size": 128256
 }

   "torch_dtype": "bfloat16",
   "transformers_version": "4.51.0",
   "use_cache": false,
+  "vocab_size": 128256,
+  "use_bidirectional_attention": true,
 }