使用没有 IPyWidgets 的 Huggingface Transformer

Question

我正在尝试在名为 Deepnote 的托管 Jupyter 笔记本平台中使用 Huggingface Transformers 库。我想通过管道类下载模型，但不幸的是，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更改日志级别，这是不可能将其设置为logging.NOTSET其中需要使用跳过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')

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