sentence-transformers How to use benjamintli/modernbert-cosqa with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("benjamintli/modernbert-cosqa")
sentences = [
"python create path if doesnt exist",
"def clean_whitespace(string, compact=False):\n \"\"\"Return string with compressed whitespace.\"\"\"\n for a, b in (('\\r\\n', '\\n'), ('\\r', '\\n'), ('\\n\\n', '\\n'),\n ('\\t', ' '), (' ', ' ')):\n string = string.replace(a, b)\n if compact:\n for a, b in (('\\n', ' '), ('[ ', '['),\n (' ', ' '), (' ', ' '), (' ', ' ')):\n string = string.replace(a, b)\n return string.strip()",
"def rotateImage(img, angle):\n \"\"\"\n\n querries scipy.ndimage.rotate routine\n :param img: image to be rotated\n :param angle: angle to be rotated (radian)\n :return: rotated image\n \"\"\"\n imgR = scipy.ndimage.rotate(img, angle, reshape=False)\n return imgR",
"def check_create_folder(filename):\n \"\"\"Check if the folder exisits. If not, create the folder\"\"\"\n os.makedirs(os.path.dirname(filename), exist_ok=True)"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
