小编And*_* NR的帖子

从 Hugging Face 模型加载权重时出错

我正在使用变压器并且我已经加载了一个模型并且它工作正常:

from transformers import AutoModelForSequenceClassification
from transformers import AutoTokenizer

task='sentiment'
MODEL = "cardiffnlp/twitter-roberta-base-{task}"
tokenizer = AutoTokenizer.from_pretrained(MODEL)

# PT
model = AutoModelForSequenceClassification.from_pretrained(MODEL) 
model.save_pretrained(MODEL)
Run Code Online (Sandbox Code Playgroud)

但是如果我尝试加载另一个任务,例如“情感”或“仇恨”,我会收到此错误:

from transformers import AutoModelForSequenceClassification
from transformers import AutoTokenizer

task='emotion'
MODEL = "cardiffnlp/twitter-roberta-base-{task}"
tokenizer = AutoTokenizer.from_pretrained(MODEL)

# PT
model = AutoModelForSequenceClassification.from_pretrained(MODEL)  ## Here I get the error
model.save_pretrained(MODEL)
Run Code Online (Sandbox Code Playgroud)

这个错误:

OSError: Can't load weights for 'cardiffnlp/twitter-roberta-base-emotion'. Make sure that:

- 'cardiffnlp/twitter-roberta-base-emotion' is a correct model identifier listed on 'https://huggingface.co/models'

- or 'cardiffnlp/twitter-roberta-base-emotion' is the correct path to a directory …
Run Code Online (Sandbox Code Playgroud)

python error-handling model huggingface-transformers

2
推荐指数
1
解决办法
4287
查看次数