RS
Size: a a a
RS
N
N
B
OS
N
N
N
N
Eg
AP
import torchДалее, пользуясь библиотекой sbert, делаю:
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("cointegrated/LaBSE-en-ru")
model = AutoModel.from_pretrained("cointegrated/LaBSE-en-ru")
sentences = ["Hello World", "Привет Мир"]
encoded_input = tokenizer(sentences, padding=True, truncation=True, max_length=64, return_tensors='pt')
with torch.no_grad():
model_output = model(**encoded_input)
embeddings = model_output.pooler_output
embeddings = torch.nn.functional.normalize(embeddings)
from sentence_transformers import SentenceTransformerДалее, сравниваю получившиеся вложения:
model = SentenceTransformer('cointegrated/LaBSE-en-ru')
sbert_embeddings = model.encode(sentences)
torch.allclose(embeddings[0], torch.tensor(sbert_embeddings[0]))Выдает False. Можете, пожалуйста, подсказать, с чем это может быть связано?
SМ
AP
IS
AP
SМ
SМ