预训练roberta关系提取属性误差

Question

我正在尝试让以下经过预先训练的拥抱脸模型开始工作：https://huggingface.co/mmoradi/Robust-Biomed-RoBERTa-RelationClassification

我使用以下代码：

from transformers import AutoTokenizer, AutoModel
  
tokenizer = AutoTokenizer.from_pretrained("mmoradi/Robust-Biomed-RoBERTa-RelationClassification")

model = AutoModel.from_pretrained("mmoradi/Robust-Biomed-RoBERTa-RelationClassification")

inputs = tokenizer("""The colorectal cancer was caused by mutations in angina""")
outputs = model(**inputs)

由于某些原因，在试图生成输出时会出现以下错误，因此在我的代码的最后一行：

-> 796 input_shape = input_ids.size() 797 elif inputs_embeds不是None: 798 input_shape = inputs_embeds.size():-1

AttributeError：'list‘对象没有属性'size’

输入如下：

{'input_ids': [0, 133, 11311, 1688, 3894, 337, 1668, 21, 1726, 30, 28513, 11, 1480, 347, 2], 'attention_mask': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]}

我不知道如何去调试这个，所以任何帮助或提示都是欢迎的！

Answer 1

您必须指定您想要的张量类型，以作为tokenizer的回报。如果没有，它将返回一个包含两个列表(input_ids和attention_mask)的字典：

inputs = tokenizer("""The colorectal cancer was caused by mutations in angina""", return_tensors="pt")