all-mpnet-base-v2-ds-rag-17g

pip install -U sentence-transformers

from sentence_transformers import SentenceTransformer

# Download from the 🤗 Hub
model = SentenceTransformer("sentence_transformers_model_id")
# Run inference
sentences = [
    'In doing this, there are three decisions we must make:\n\n1.',
    'g(•) defines the hypothesis class H , and a particular value of θ instantiates one hypothesis h ∈ H .',
    'Although replacing traces (Section 7.8) are known to have advantages in tabular methods, replacing traces do not directly extend to the use of function approximation.',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]

# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities)
# tensor([[1.0000, 0.3814, 0.1328],
#         [0.3814, 1.0000, 0.1478],
#         [0.1328, 0.1478, 1.0000]])