nvidia
/

llama-embed-nemotron-8b

Feature Extraction

sentence-transformers

sentence-similarity

Model card Files Files and versions

Update config.json

#12

by baseten-admin - opened Jan 6

base: refs/heads/main

←

from: refs/pr/12

Discussion Files changed

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
 }