Pooling fix a7e25c2 verified
Boris Malashenko commited on
How to use BorisTM/bge-m3_en_ru with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("BorisTM/bge-m3_en_ru")
sentences = [
"Это счастливый человек",
"Это счастливая собака",
"Это очень счастливый человек",
"Сегодня солнечный день"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]How to use BorisTM/bge-m3_en_ru with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("BorisTM/bge-m3_en_ru")
model = AutoModel.from_pretrained("BorisTM/bge-m3_en_ru", device_map="auto")