Closed feb-cloud closed 3 years ago
ghosthamlet
commented
3 years ago 本库的gpt2模型结构和通用的不太一样，你需要从本库import修改后的模型：
from gpt2_ml_torch.modeling_gpt2 import GPT2LMHeadModel
输入输出和通用gpt2是兼容的，用法参考：https://github.com/ghosthamlet/gpt2-ml-torch/blob/master/gpt2_ml_torch/generate.py
feb-cloud
commented
3 years ago 本库的gpt2模型结构和通用的不太一样，你需要从本库import修改后的模型：
from gpt2_ml_torch.modeling_gpt2 import GPT2LMHeadModel输入输出和通用gpt2是兼容的，用法参考：https://github.com/ghosthamlet/gpt2-ml-torch/blob/master/gpt2_ml_torch/generate.py
非常感谢您的回复，辛苦啦。
ghosthamlet
commented
3 years ago 不客气
作者您好，我把torch模型拿到gpt2-chinese这个项目中使用，发现生成的文章都是乱码的，不清楚哪里出了问题，希望能够得到您的指点，多谢！期待您的回复。