Python 导入错误：无法导入名称'；带有LMHEAD和x27的自动建模；从'；变压器'；

这就是我试图运行的所有代码：

from transformers import AutoModelWithLMHead, AutoTokenizer
import torch

tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-small")
model = AutoModelWithLMHead.from_pretrained("microsoft/DialoGPT-small")

我得到这个错误：

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-14-aad2e7a08a74> in <module>
----> 1 from transformers import AutoModelWithLMHead, AutoTokenizer
      2 import torch
      3 
      4 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-small")
      5 model = AutoModelWithLMHead.from_pretrained("microsoft/DialoGPT-small")

ImportError: cannot import name 'AutoModelWithLMHead' from 'transformers' (c:\python38\lib\site-packages\transformers\__init__.py)

---------------------------------------------------------------------------
ImportError回溯（最近一次呼叫最后一次）
在里面
---->1从transformers进口带LMHEAD的Automodel，AutoTokenizer
2进口火炬
3.
4标记器=来自预训练（“microsoft/DialoGPT small”）的自动烹饪器
5 model=AutoModelWithLMHead.from_pretrained（“microsoft/DialoGPT small”）
ImportError:无法从“transformers”（c:\python38\lib\site packages\transformers\\uuu init\uuuu.py）导入名称“AutoModelWithlHead”

---

我该怎么办？

我解决了！显然，我的版本中删除了带有LMHEAD的AutoModel

现在，您需要对因果语言模型使用

AutoModelForCausalLM

，对屏蔽语言模型使用

automodelforaskedlm

，对编码器-解码器模型使用

automodelforseqseqlm

在我的例子中，代码如下所示：

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-small")
model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-small")

---

你试过从变形金刚库导入其他东西吗？你使用的是什么版本的变形金刚？@cronoik我使用的是最新版本。变压器-3.0。2@A.T.B导入自动烹饪器效果很好…我无法复制这个。请运行

import transformers

和

transformers.\uuuu版本\uuuuu

以确保您真正使用的是

3.0.2

。你是如何安装的？