Sentence Similarity
sentence-transformers
PyTorch
TensorFlow
ONNX
Safetensors
OpenVINO
Transformers
roberta
feature-extraction
text-embeddings-inference
Instructions to use sentence-transformers/stsb-roberta-base 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 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("sentence-transformers/stsb-roberta-base") 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 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/stsb-roberta-base") model = AutoModel.from_pretrained("sentence-transformers/stsb-roberta-base", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
Add exported openvino model 'openvino_model_qint8_quantized.xml' fb8c0e7 verified
Add new SentenceTransformer model with an openvino backend b45158d verified
Add new SentenceTransformer model with an openvino backend 8e7fb99 verified
Add exported ONNX model 'model_qint8_avx512_vnni.onnx' 8eb109a verified
Add exported ONNX model 'model_qint8_avx512.onnx' 12d5bbb verified
Add exported ONNX model 'model_quint8_avx2.onnx' eace820 verified
Add exported ONNX model 'model_qint8_arm64.onnx' 8a8069a verified
Add exported ONNX model 'model_O4.onnx' 50a7427 verified
Add exported ONNX model 'model_O3.onnx' 51df4d2 verified
Add exported ONNX model 'model_O2.onnx' 5b651c9 verified
Add exported ONNX model 'model_O1.onnx' b8e813e verified
Add new SentenceTransformer model with an onnx backend c7e8127 verified
Add TF weights 20afd58
Joao Gante commited on
Update README.md f8e6f72
Update README.md 1ad2f87
Add new SentenceTransformer model. 4c0755c
nreimers commited on