python - 使用不带 IPyWidgets 的拥抱脸转换器

Question

我正在尝试在名为 Deepnote 的托管 Jupyter 笔记本平台中使用 huggingface 转换器库。我想通过管道类下载模型，但不幸的是 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 不是一个选项

score 6 · Accepted Answer

您必须禁用变压器日志记录。尽管可以使用transformers.logging.set_verbosity来更改日志级别，但无法将其设置logging.NOTSET为跳过使用and所需的级别。所以我们需要像这样破解它：IProgresstqdm

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')

查看我的 Deepnote 项目了解详情；）

1 回答 1