How to use Den4ikAI/rubert-tiny2-retriever with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("Den4ikAI/rubert-tiny2-retriever") sentences = [ "query: Когда родился Пушкин?", "passage: Алекса́ндр Серге́евич Пу́шкин (26 мая [6 июня] 1799, Москва — 29 января [10 февраля] 1837, Санкт-Петербург) — русский поэт, драматург и прозаик, заложивший основы русского реалистического направления[2], литературный критик[3] и теоретик литературы, историк[3], публицист, журналист[3].", "passage: Пушкин ловил кайф со своими друзьями", "passage: Пушкин из самых авторитетных литературных деятелей первой трети XIX века. Ещё при жизни Пушкина сложилась его репутация величайшего национального русского поэта[4][5]. Пушкин рассматривается как основоположник современного русского литературного языка[~ 2]." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
How to use Den4ikAI/rubert-tiny2-retriever with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("Den4ikAI/rubert-tiny2-retriever") model = AutoModel.from_pretrained("Den4ikAI/rubert-tiny2-retriever")
SFconvertbot