运行代码发现错误如下:在这里插入图片描述
解决:
将导入模块

from transformers.modeling_gpt2 import GPT2LMHeadModel

改为:

from transformers.models.gpt2.modeling_gpt2 import GPT2LMHeadModel
Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐