Sentence Similarity
sentence-transformers
PyTorch
TensorFlow
JAX
ONNX
Safetensors
OpenVINO
Transformers
bert
feature-extraction
text-embeddings-inference
Instructions to use sentence-transformers/bert-base-nli-cls-token with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use sentence-transformers/bert-base-nli-cls-token with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("sentence-transformers/bert-base-nli-cls-token") 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/bert-base-nli-cls-token with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/bert-base-nli-cls-token") model = AutoModel.from_pretrained("sentence-transformers/bert-base-nli-cls-token", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Commit History
Add TF weights e843bdc
Joao Gante commited on
Update README.md 5b0c02b
Update README.md ce84633
Add new SentenceTransformer model. 5d079c4
nreimers commited on
Update README.md e1f20cf
Omar Sanseviero commited on