无法从变压器导入 BertModel

Question

我正在尝试从转换器中导入 BertModel，但它失败了。这是我正在使用的代码

from transformers import BertModel, BertForMaskedLM

这是我得到的错误

ImportError: cannot import name 'BertModel' from 'transformers'

谁能帮我解决这个问题？

Answer 1

修正了错误。这是代码

from transformers.modeling_bert import BertModel, BertForMaskedLM

Answer 2

您也可以使用您的代码from transformers import BertModel, BertForMaskedLM；只需确保您的 transformers 已更新。

Answer 3

最新版本的变形金刚已修复此问题

pip install upgrade transformers