How to use johnnyboycurtis/ModernBERT-small-sts with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("johnnyboycurtis/ModernBERT-small-sts") sentences = [ "A modern bathtub in a bathroom is displayed.", "Different types of tiles are on the walls, floor and tub.", "A man sitting on a park bench looking towards a fountain and sculpture.", "A bathroom with a shower and his and her sinks." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
