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

from transformers.modeling_gpt2 import GPT2LMHeadModel

改为:

from transformers.models.gpt2.modeling_gpt2 import GPT2LMHeadModel
Logo

欢迎加入DeepSeek 技术社区。在这里,你可以找到志同道合的朋友,共同探索AI技术的奥秘。

更多推荐