使用不带IPyWidgets的拥抱面变压器

Question

我正试图在名为Deepnote的Jupyter笔记本托管平台中使用huggingface transformers库。我想通过pipeline类下载一个模型，但不幸的是deepnote不支持IPyWidgets。在使用transformer时，有没有办法禁用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.loggingset_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项目以了解详细信息；(