python - 无法从变压器导入管道

Question

我已经安装pytorch了 withconda和transformerswith pip。

我可以import transformers没有问题，但是当我尝试import pipeline from transformers遇到异常时：

from transformers import pipeline
---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-4-69a9fd07ccac> in <module>
----> 1 from transformers import pipeline

ImportError: cannot import name 'pipeline' from 'transformers' (C:\Users\Alienware\Anaconda3\envs\tf2\lib\site-packages\transformers\__init__.py)

这是它搜索init .py 文件的目录的视图：

是什么导致了问题，我该如何解决？

Answer 1

检查transformers版本。确保你是最新的。管道是最近才引入的，您可能有旧版本。

Answer 2

5 月底宣布spacy-transformers v0.6.0与transformers v2.5.0兼容。所以，如果你也打算使用 spacy-transformers，最好使用 v2.5.0 来代替最新版本。

所以，试试吧；

pip install transformers==2.5.0

pip install spacy-transformers==0.6.0

并同时使用 2 个预训练模型，没有任何问题