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 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