使用Transformers Pipeline做情感分析时遇导入错误求助

问题原因 #

这个报错是transformers库和它的依赖库（比如accelerate）版本不兼容导致的——add_code_sample_docstrings函数的参数在不同版本里有变动，调用时出现了参数不匹配的情况。

具体解决步骤 #

• 先彻底卸载相关冲突库：

  pip uninstall -y transformers accelerate tokenizers
  

• 安装经过验证的兼容版本组合：

  pip install transformers==4.34.0 accelerate==0.23.0 tokenizers==0.14.1
  

• 重新运行你的代码，就能正常初始化情感分析pipeline了。

备选方案 #

如果不想调整版本，也可以直接指定情感分析专用的轻量模型，绕开GPT2相关的加载问题：

from transformers import pipeline

# 指定distilbert的情感分析预训练模型
sent_pipeline = pipeline('sentiment-analysis', model='distilbert-base-uncased-finetuned-sst-2-english')

内容的提问来源于stack exchange，提问作者ngbx05