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:
# 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") - Notebooks
- Google Colab
- Kaggle
remove
Browse files- onnx_model/model.onnx +0 -3
- onnx_model/optimized_model.onnx +0 -3
onnx_model/model.onnx
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:c26b1b3e2f210b1e49e7a1a5067f6451c7c9039ac5a91848c5d455372fd5d618
|
| 3 |
-
size 17532461
|
|
|
|
|
|
|
|
|
|
|
|
onnx_model/optimized_model.onnx
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:c26b1b3e2f210b1e49e7a1a5067f6451c7c9039ac5a91848c5d455372fd5d618
|
| 3 |
-
size 17532461
|
|
|
|
|
|
|
|
|
|
|
|