sentence-transformers How to use ciCic/distilbert-base-uncased-sts-matryoshka with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("ciCic/distilbert-base-uncased-sts-matryoshka")
sentences = [
"The cat is watching out of the window",
"That is a happy dog",
"That is a very funny person",
"A cat stands by the window and looks outside"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]
