Add new SentenceTransformer model with an onnx backend
#1
by
T3LS
- opened
Hello!
This pull request has been automatically generated from the push_to_hub
method from the Sentence Transformers library.
Full Model Architecture:
SentenceTransformer(
(0): StaticEmbedding(
(embedding): EmbeddingBag(105879, 32, mode='mean')
)
)
Tip:
Consider testing this pull request before merging by loading the model from this PR with the revision
argument:
from sentence_transformers import SentenceTransformer
# TODO: Fill in the PR number
pr_number = 2
model = SentenceTransformer(
"T3LS/static-similarity-mrl-multilingual-v1-32d-fp16",
revision=f"refs/pr/{pr_number}",
backend="onnx",
)
# Verify that everything works as expected
embeddings = model.encode(["The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium."])
print(embeddings.shape)
similarities = model.similarity(embeddings, embeddings)
print(similarities)