Sentence Similarity
sentence-transformers
Safetensors
Transformers
ministral3
feature-extraction
text
text-embeddings
retrieval
semantic-search
rag
vllm
Instructions to use nvidia/Nemotron-3-Embed-1B-BF16 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use nvidia/Nemotron-3-Embed-1B-BF16 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("nvidia/Nemotron-3-Embed-1B-BF16") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Transformers
How to use nvidia/Nemotron-3-Embed-1B-BF16 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("nvidia/Nemotron-3-Embed-1B-BF16") model = AutoModel.from_pretrained("nvidia/Nemotron-3-Embed-1B-BF16") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -116,7 +116,7 @@ The resulting 2B model was then distilled from the fine-tuned [Nemotron-3-Embed-
|
|
| 116 |
|
| 117 |
**Output Parameters:** One-Dimensional (1D) embedding vector per input text string <br>
|
| 118 |
|
| 119 |
-
**Other Properties Related to Output:** The model outputs a 2048-dimensional embedding vector for each input text string. <br>
|
| 120 |
|
| 121 |
Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA's hardware (e.g. GPU cores) and software frameworks (e.g., CUDA libraries), the model achieves faster training and inference times compared to CPU-only solutions. <br>
|
| 122 |
|
|
|
|
| 116 |
|
| 117 |
**Output Parameters:** One-Dimensional (1D) embedding vector per input text string <br>
|
| 118 |
|
| 119 |
+
**Other Properties Related to Output:** The model outputs a 2048-dimensional embedding vector for each input text string. It also supports dynamic embedding sizes by slicing the vector from the start (for example, keeping the first 1024 or 512 dimensions). These sliced embeddings remain highly functional, provided the resulting sub-vector is re-normalized (L2 normalization) after slicing. <br>
|
| 120 |
|
| 121 |
Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA's hardware (e.g. GPU cores) and software frameworks (e.g., CUDA libraries), the model achieves faster training and inference times compared to CPU-only solutions. <br>
|
| 122 |
|