How to use PyxiLab/Pyx-embeds with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("PyxiLab/Pyx-embeds") sentences = [ "search_query: i love autotrain", "search_query: huggingface auto train", "search_query: hugging face auto train", "search_query: i love autotrain" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
