Monor improvement 7982df2
Oleg commited on
How to use geologist387/BERTA-transformed with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("geologist387/BERTA-transformed")
sentences = [
"Это счастливый человек",
"Это счастливая собака",
"Это очень счастливый человек",
"Сегодня солнечный день"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]How to use geologist387/BERTA-transformed with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("geologist387/BERTA-transformed", dtype="auto")