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]
