new samples in training dataset fa4f459
koziev ilya commited on
How to use inkoziev/sbert_synonymy with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("inkoziev/sbert_synonymy")
sentences = [
"Кошка ловит мышку",
"Мышка преследуема кошкой",
"Кошка гонится за мышью",
"Кошка ловит кайф"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]How to use inkoziev/sbert_synonymy with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("inkoziev/sbert_synonymy")
model = AutoModel.from_pretrained("inkoziev/sbert_synonymy", device_map="auto")