sentence-transformers How to use stefan-jo/mxbai-edge-colbert-v0-32m-fiqa-kmeans-pf1-6 with sentence-transformers:
from pylate import models
queries = [
"Which planet is known as the Red Planet?",
"What is the largest planet in our solar system?",
]
documents = [
["Mars is the Red Planet.", "Venus is Earth's twin."],
["Jupiter is the largest planet.", "Saturn has rings."],
]
model = models.ColBERT(model_name_or_path="stefan-jo/mxbai-edge-colbert-v0-32m-fiqa-kmeans-pf1-6")
queries_emb = model.encode(queries, is_query=True)
docs_emb = model.encode(documents, is_query=False) 
