无法从变压器导入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已更新。