superduper_sentence_transformers

superduper allows users to work with self-hosted embedding models via Sentence-Transformers.

Installation

pip install superduper_sentence_transformers

API

• Code
• API-docs

Class	Description
superduper_sentence_transformers.model.SentenceTransformer	A model for sentence embeddings using sentence-transformers.

Examples

SentenceTransformer

from superduper import vector
from superduper_sentence_transformers import SentenceTransformer
import sentence_transformers
model = SentenceTransformer(
    identifier="embedding",
    object=sentence_transformers.SentenceTransformer("BAAI/bge-small-en"),
    datatype=vector(shape=(1024,)),
    postprocess=lambda x: x.tolist(),
    predict_kwargs={"show_progress_bar": True},
)
model.predict("What is superduper")