Sentence Similarity
sentence-transformers
Safetensors
Transformers
bert
feature-extraction
text-embeddings-inference
Instructions to use optimum-internal-testing/sentence-transformers-stsb-bert-tiny with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use optimum-internal-testing/sentence-transformers-stsb-bert-tiny with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("optimum-internal-testing/sentence-transformers-stsb-bert-tiny") 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 optimum-internal-testing/sentence-transformers-stsb-bert-tiny with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("optimum-internal-testing/sentence-transformers-stsb-bert-tiny") model = AutoModel.from_pretrained("optimum-internal-testing/sentence-transformers-stsb-bert-tiny", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Adding ONNX file of this model
Verified signatureBeep boop I am the [ONNX export bot 🤖🏎️](https://huggingface.co/spaces/onnx/export). On behalf of [echarlaix](https://huggingface.co/echarlaix), I would like to add to this repository the model converted to ONNX. What is ONNX? It stands for "Open Neural Network Exchange", and is the most commonly used open standard for machine learning interoperability. You can find out more at [onnx.ai](https://onnx.ai/)! The exported ONNX model can be then be consumed by various backends as TensorRT or TVM, or simply be used in a few lines with 🤗 Optimum through ONNX Runtime, check out how [here](https://huggingface.co/docs/optimum/main/en/onnxruntime/usage_guides/models)!e759741 -
add model
ca01eb6
-
remove
58747b7 -
rename
ddea5c4 -
add onnx model
64d5957 -
initial commit
Verified signaturefa299c9