update README 48cc6f4 unverified
Roman Solomatin commited on
How to use sergeyzh/rubert-tiny-turbo with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("sergeyzh/rubert-tiny-turbo")
sentences = [
"Это счастливый человек",
"Это счастливая собака",
"Это очень счастливый человек",
"Сегодня солнечный день"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]How to use sergeyzh/rubert-tiny-turbo with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("sergeyzh/rubert-tiny-turbo")
model = AutoModel.from_pretrained("sergeyzh/rubert-tiny-turbo")