sentence-transformers How to use leonweber/checkpoints with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("leonweber/checkpoints")
sentences = [
"<start> FTYGHYHHYHGGTTGRREEEEEEEEDEEEE <end>",
"on",
"later",
"The"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]