Sentence Similarity
sentence-transformers
PyTorch
TensorFlow
JAX
ONNX
Safetensors
OpenVINO
Transformers
roberta
feature-extraction
text-embeddings-inference
Instructions to use sentence-transformers/stsb-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/stsb-roberta-base-v2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("sentence-transformers/stsb-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/stsb-roberta-base-v2 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/stsb-roberta-base-v2") model = AutoModel.from_pretrained("sentence-transformers/stsb-roberta-base-v2") - Inference
- Notebooks
- Google Colab
- Kaggle
Commit History
Add exported openvino model 'openvino_model_qint8_quantized.xml' da156ad verified
Add new SentenceTransformer model with an openvino backend 1ffaed3 verified
Add new SentenceTransformer model with an openvino backend 6545587 verified
Add exported ONNX model 'model_qint8_avx512_vnni.onnx' 583c568 verified
Add exported ONNX model 'model_qint8_avx512.onnx' afc610c verified
Add exported ONNX model 'model_quint8_avx2.onnx' 157ef86 verified
Add exported ONNX model 'model_qint8_arm64.onnx' d7bb6c4 verified
Add exported ONNX model 'model_O4.onnx' 4a41b5e verified
Add exported ONNX model 'model_O3.onnx' 147e6dd verified
Add exported ONNX model 'model_O2.onnx' 9fa6297 verified
Add exported ONNX model 'model_O1.onnx' f88ac48 verified
Add TF weights b5e9e8d
Joao Gante commited on
Update README.md 5a480fa
Add new SentenceTransformer model. c9ac54c
nreimers commited on
upload flax model 3f1a150
allow flax e6dff75
upload 3d1ab8d
nreimers commited on