BLEURT-20-D12

Instala el modelo con:

pip install git+https://github.com/lucadiliello/bleurt-pytorch.git

Carga el modelo y realiza predicciones con el siguiente código:

import torch
from bleurt_pytorch import BleurtConfig, BleurtForSequenceClassification, BleurtTokenizer

config = BleurtConfig.from_pretrained('lucadiliello/BLEURT-20-D12')
model = BleurtForSequenceClassification.from_pretrained('lucadiliello/BLEURT-20-D12')
tokenizer = BleurtTokenizer.from_pretrained('lucadiliello/BLEURT-20-D12')

references = ["un pájaro canta junto a la ventana", "esta es una frase aleatoria"]
candidates = ["un pájaro canta junto a la ventana", "esto parece una frase aleatoria"]

model.eval()
with torch.no_grad():
    inputs = tokenizer(references, candidates, padding='longest', return_tensors='pt')
    res = model(**inputs).logits.flatten().tolist()
    print(res)
# [0.9604414105415344, 0.8080050349235535]