Sentence Similarity
sentence-transformers
PyTorch
TensorFlow
JAX
ONNX
Safetensors
OpenVINO
Transformers
roberta
feature-extraction
text-embeddings-inference
Instructions to use sentence-transformers/msmarco-roberta-base-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use sentence-transformers/msmarco-roberta-base-v2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("sentence-transformers/msmarco-roberta-base-v2") sentences = [ "That is a happy person", "That is a happy dog", "That is a very happy person", "Today is a sunny day" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Transformers
How to use sentence-transformers/msmarco-roberta-base-v2 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/msmarco-roberta-base-v2") model = AutoModel.from_pretrained("sentence-transformers/msmarco-roberta-base-v2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
Add TF weights b23dfbc
Joao Gante commited on
Update README.md 5703c16
Add new SentenceTransformer model. 669b432
nreimers commited on
upload flax model 9cf6919
allow flax c823e1f
upload 5ef767d
nreimers commited on