Feature Extraction
sentence-transformers
Safetensors
Transformers
multilingual
llama_bidirec
text
sentence-similarity
mteb
mmteb
custom_code
text-embeddings-inference
Instructions to use nvidia/llama-embed-nemotron-8b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use nvidia/llama-embed-nemotron-8b with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("nvidia/llama-embed-nemotron-8b", trust_remote_code=True) 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/llama-embed-nemotron-8b with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="nvidia/llama-embed-nemotron-8b", trust_remote_code=True)# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("nvidia/llama-embed-nemotron-8b", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
#4
by ybabakhin - opened
README.md
CHANGED
|
@@ -25,13 +25,9 @@ This model is for non-commercial/research use only.
|
|
| 25 |
|
| 26 |
### License/Terms of Use
|
| 27 |
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
Governing Terms for `llama-embed-nemotron-8b` model: [Customized NSCLv1](https://huggingface.co/nvidia/llama-embed-nemotron-8b/blob/main/LICENSE) <br>
|
| 31 |
Additional Information: [Llama-3.1 Community License Agreement](https://huggingface.co/meta-llama/Llama-3.1-70B-Instruct/blob/main/LICENSE) for meta-llama/Llama-3.1-8B. [Acceptable Use Policy](https://llama.meta.com/llama3_1/use-policy). Built with Llama.
|
| 32 |
|
| 33 |
-
For each dataset a user elects to use, the user is responsible for checking if the dataset license is fit for the intended purpose.
|
| 34 |
-
|
| 35 |
### Team
|
| 36 |
|
| 37 |
- Yauhen Babakhin
|
|
|
|
| 25 |
|
| 26 |
### License/Terms of Use
|
| 27 |
|
| 28 |
+
Governing Terms for `llama-embed-nemotron-8b` model: [NVIDIA License](https://huggingface.co/nvidia/llama-embed-nemotron-8b/blob/main/LICENSE) <br>
|
|
|
|
|
|
|
| 29 |
Additional Information: [Llama-3.1 Community License Agreement](https://huggingface.co/meta-llama/Llama-3.1-70B-Instruct/blob/main/LICENSE) for meta-llama/Llama-3.1-8B. [Acceptable Use Policy](https://llama.meta.com/llama3_1/use-policy). Built with Llama.
|
| 30 |
|
|
|
|
|
|
|
| 31 |
### Team
|
| 32 |
|
| 33 |
- Yauhen Babakhin
|