tensorflow 来自Hugging-Face的变形金刚模型抛出无法加载特定类的错误

z5btuh9x 于 2022-12-27 发布在其他

关注(0)|答案(3)|浏览(297)

嗨，运行下面的代码后，我得到以下错误.

值错误：无法使用以下任何类加载模型facebook/bart-large-mnli：（〈类'变压器.模型.自动.建模_tf_auto. TFAutoModelForSequenceClassification'〉，）.*

import tensorflow as tf
from transformers import pipeline

classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")

有人能帮帮忙吗。谢谢！

3条答案

我也遇到了同样的问题!有人在这里评论说你需要安装PyTorch（https://github.com/huggingface/transformers/issues/16849）。
总结一下：

有些型号仅作为PyTorch型号存在（例如deepset/roberta-base-squad2）。
调用pipeline()将根据您的机器上安装的内容（在我的例子中是venv）选择框架（TF或PyTorch）
如果两者都安装了，则会选择Torch
如果你没有安装PyTorch，它会抛出上述错误
- - 安装PyTorch为我解决了这个问题!**
在GitHub问题中，提到了另一种解决方案：使用from_pt=True在TF中加载模型，并使用save_pretrained和push_to_hub将其另存为TF模型的个人副本

请使用以下内容：

!pip install pytorch-pretrained-bert
import pytorch_pretrained_bert as ppb
assert 'bert-large-cased' in ppb.modeling.PRETRAINED_MODEL_ARCHIVE_MAP

现在运行原始代码。

在我的例子中，问题是由非常慢的网速引起的。值得注意的是，如果在下载过程中不断失败，您可以重试，直到它工作（希望如此）。