sentence-transformers How to use benjamintli/modernbert-code with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("benjamintli/modernbert-code")
sentences = [
"@Override\n public void encode(final OtpOutputStream buf) {\n final int arity = elems.length;\n\n buf.write_tuple_head(arity);\n\n for (int i = 0; i < arity; i++) {\n buf.write_any(elems[i]);\n }\n }",
"fetch function with the same interface than in cozy-client-js",
"Convert this tuple to the equivalent Erlang external representation.\n\n@param buf\nan output stream to which the encoded tuple should be written.",
"Delete a customer by it's id.\n\n@param int $id The id\n\n@return bool\n@throws \\Throwable in case something went wrong when deleting."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
