设置 Visual Studio Code 以从 Hugging Face 运行模型

Question

我正在尝试从拥抱脸导入模型并在 Visual Studio Code 中使用它们。 我安装了变压器 tensorflow 和手电筒。 我曾尝试在网上查看多个教程，但一无所获。 我正在尝试运行以下代码：

from transformers import pipeline
classifier = pipeline('sentiment-analysis')
result = classifier("I hate it when I'm sitting under a tree and an apple hits my head.")
print(result)

但是，我收到以下错误：

No model was supplied, defaulted to distilbert-base-uncased-finetuned-sst-2-english and revision af0f99b (https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english).
Using a pipeline without specifying a model name and revision in production is not recommended.
Traceback (most recent call last):
  File "c:\Users\user\Desktop\Artificial Intelligence\transformers\Workshops\workshop_3.py", line 4, in <module>
    classifier = pipeline('sentiment-analysis')
  File "C:\Users\user\Desktop\Artificial Intelligence\transformers\src\transformers\pipelines\__init__.py", line 702, in pipeline
    framework, model = infer_framework_load_model(
  File "C:\Users\user\Desktop\Artificial Intelligence\transformers\src\transformers\pipelines\base.py", line 266, in infer_framework_load_model
    raise ValueError(f"Could not load model {model} with any of the following classes: {class_tuple}.")
ValueError: Could not load model distilbert-base-uncased-finetuned-sst-2-english with any of the following classes: (<class 'transformers.models.auto.modeling_auto.AutoModelForSequenceClassification'>, <class 'transformers.models.auto.modeling_tf_auto.TFAutoModelForSequenceClassification'>, <class 'transformers.models.distilbert.modeling_distilbert.DistilBertForSequenceClassification'>, <class 'transformers.models.distilbert.modeling_tf_distilbert.TFDistilBertForSequenceClassification'>).

我已经在网上搜索了设置转换器以在 Visual Studio Code 中使用的方法，但没有任何帮助。

您是否知道如何解决此错误，或者如果有人知道如何成功地将 Hugging Face 中的模型用于我的代码中，将不胜感激？

Answer 1

这个问题与 Hugging Face 本身有关，可能更多与安装和您采取的安装步骤有关（以及您的程序可能对模型自动下载到的缓存文件的访问。）。

从我所看到的：1/您的程序无法访问 model 2/您的程序在一些边缘情况下抛出特定值错误

如果 1/ 看看这里：[https://huggingface.co/docs/transformers/installation#cache-setup][1]

请注意，文档会遍历预训练模型的下载位置。 检查它是否在此处下载： C:\Users\username\.cache\huggingface\hub （当然是在您的计算机上使用您自己的用户名。检查缓存位置以确保它已下载？（您可以检查提到的缓存位置。）

其次，如果由于某种原因，下载出现问题，您可以尝试手动下载并通过离线模式进行（这更多是为了让它运行起来）： https://huggingface.co/docs/transformers/installation ＃离线模式

三、如果是下载的，是否有权限访问.cache？ （尝试以管理员身份在 Windows 终端上运行您的程序（如果它是您信任的程序）。）。 各种方法 - 找到一个你觉得舒服的方法，这里有一些来自 Stackoverflow/StackExchange 的提示： Opening up Windows Terminal with提升权限，从 Windows 终端或这个： Z5E056C500A1C4B6A71010B50D807BADE56//superuser/comquestions/107BADE550B50D807BADE5。 -windows-terminal-as-admin-with-winr

如果 2/ 我看到人们在找不到特定值（与您的不同但相似）方面提出了非常具体的问题，并且通过安装 PyTorch 解决了该问题，因为某些型号仅作为 PyTorch 型号存在。 您可以在此处查看@YokoHono 的完整回复： Hugging-Face 的 Transformers model 引发无法加载特定类的错误