Bleurt Large 128

🚀 BLEURT

這是ACL論文 "BLEURT: Learning Robust Metrics for Text Generation" 中原始BLEURT模型的Pytorch版本，該論文由Google Research的Thibault Sellam、Dipanjan Das和Ankur P. Parikh撰寫。

模型轉換的代碼源自此筆記本，該筆記本在此處被提及。

🚀 快速開始

本項目是ACL論文中原始BLEURT模型的Pytorch版本，為文本生成的評估提供了強大的工具。通過簡單的代碼調用，即可使用該模型進行文本評估。

💻 使用示例

基礎用法

from transformers import AutoModelForSequenceClassification, AutoTokenizer
import torch

tokenizer = AutoTokenizer.from_pretrained("Elron/bleurt-large-128")
model = AutoModelForSequenceClassification.from_pretrained("Elron/bleurt-large-128")
model.eval()

references = ["hello world", "hello world"]
candidates = ["hi universe", "bye world"]

with torch.no_grad():
  scores = model(**tokenizer(references, candidates, return_tensors='pt'))[0].squeeze()

print(scores) # tensor([ 0.0020, -0.6647])