在没有 IPyWidgets 的情况下使用拥抱面变换器

Question

我正在尝试在名为 Deepnote 的托管 Jupyter 笔记本平台中使用 huggingface 变形金刚库。我想通过管道 class 下载模型，但不幸的是 deepnote 不支持 IPyWidgets。使用转换器时有没有办法禁用 IPywidgets？具体如下命令。


classifier = pipeline("zero-shot-classification")

以及我收到的错误。

ImportError: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html

注意：安装 IPyWidgets 不是一个选项

Answer 1

您必须禁用转换器日志记录。即使可以使用 transformers.logging.set_verbosity to change the log level, it's not possible to set it to logging.NOTSET which is required 来跳过使用 IProgress 和 tqdm。所以我们需要像这样破解它：

import transformers
import logging
transformers.logging.get_verbosity = lambda: logging.NOTSET

# transformers.logging.get_verbosity()

之后你应该可以使用：

from transformers import pipeline
pipeline('sentiment-analysis')('we love you')

查看 my Deepnote project 了解详情 ;)

在没有 IPyWidgets 的情况下使用拥抱面变换器

Use huggingface transformers without IPyWidgets

python

jupyter-notebook

ipywidgets

huggingface-transformers

deepnote