Sentence Similarity
sentence-transformers
PyTorch
TensorFlow
ONNX
Safetensors
OpenVINO
Transformers
distilbert
feature-extraction
text-embeddings-inference
Instructions to use sentence-transformers/msmarco-distilbert-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-distilbert-base-v2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("sentence-transformers/msmarco-distilbert-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-distilbert-base-v2 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/msmarco-distilbert-base-v2") model = AutoModel.from_pretrained("sentence-transformers/msmarco-distilbert-base-v2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
Add exported openvino model 'openvino_model_qint8_quantized.xml' 2d18d92 verified
Add new SentenceTransformer model with an openvino backend 7c9fdda verified
Add new SentenceTransformer model with an openvino backend 3bd1f05 verified
Add exported ONNX model 'model_qint8_avx512_vnni.onnx' 7bff7d7 verified
Add exported ONNX model 'model_qint8_avx512.onnx' 929f437 verified
Add exported ONNX model 'model_quint8_avx2.onnx' 8f1905a verified
Add exported ONNX model 'model_qint8_arm64.onnx' ac3635b verified
Add exported ONNX model 'model_O4.onnx' c901bf2 verified
Add exported ONNX model 'model_O3.onnx' b56d631 verified
Add exported ONNX model 'model_O2.onnx' 18268e4 verified
Add exported ONNX model 'model_O1.onnx' 3cb1dcb verified
Add new SentenceTransformer model with an onnx backend 7361d3d verified
Add TF weights f948558
Joao Gante commited on
Update README.md 46ce88f
Add new SentenceTransformer model. 7f39727
nreimers commited on
up 0259fdc
nreimers commited on
upload ba16bf3
nreimers commited on