sentence-transformers How to use LamaDiab/MiniLM-256BATCH-V6Data-SemanticEngine with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("LamaDiab/MiniLM-256BATCH-V6Data-SemanticEngine")
sentences = [
"cream 21 baby oil with almond oil",
"hi, barbie! bundle",
"nourishing baby oil",
"material: wooden. size: 15 x 30 cm."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]