sentence-transformers How to use NeuML/pylate-bert-tiny 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="NeuML/pylate-bert-tiny")
queries_emb = model.encode(queries, is_query=True)
docs_emb = model.encode(documents, is_query=False) 
