Sentence Similarity
sentence-transformers
PyTorch
Transformers
Russian
bert
feature-extraction
text-embeddings-inference
Instructions to use Den4ikAI/rubert-tiny2-retriever with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
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] - Transformers
How to use Den4ikAI/rubert-tiny2-retriever with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("Den4ikAI/rubert-tiny2-retriever") model = AutoModel.from_pretrained("Den4ikAI/rubert-tiny2-retriever", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Adding `safetensors` variant of this model
Verified signatureThis is an automated PR created with https://huggingface.co/spaces/safetensors/convert This new file is equivalent to `pytorch_model.bin` but safe in the sense that no arbitrary code can be put into it. These files also happen to load much faster than their pytorch counterpart: https://colab.research.google.com/github/huggingface/notebooks/blob/main/safetensors_doc/en/speed.ipynb The widgets on your model page will run using this model even if this is not merged making sure the file actually works. If you find any issues: please report here: https://huggingface.co/spaces/safetensors/convert/discussions Feel free to ignore this PR.56a0618
-
Update README.md
697c839
-
Upload 13 files
9176d09 -
Upload 10 files
8801347 -
Upload 12 files
a9d7c81 -
Upload 11 files
7cb8f40 -
initial commit
db64255