How to use from the
Use from the
sentence-transformers library
from sentence_transformers import SentenceTransformer

model = SentenceTransformer("NeuML/bert-tiny-prompts")

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]

BERT Tiny Prompts

This model is a copy of stsb-bert-tiny-safetensors with prompts added for testing. It's intended for use with unit tests.

Downloads last month
1,367
Safetensors
Model size
4.39M params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support