siyuyuan/KPCE

prompt_code/model.py 文件似乎有一些问题。

Ferliuc opened this issue · 1 comments

model.py 文件里的Prompt_MRCModel类下的generate_default_inputs()函数里有一行好像写错了?
原代码为:
input_embeds = tensor.cat((topic_embeds,raw_embeds),1)
修改为:
input_embeds = torch.cat((topic_embeds,raw_embeds),1)
才可正常运行。

好滴,我改一下~谢谢你:)