训练完ChatGLM后想执行推理时,发现模型报错

RuntimeError: probability tensor contains either `inf`, `nan` or element < 0

解决方法:
删除ptuning下的main.py文件第387行和396行的do_sample=True红框语句。
在这里插入图片描述

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐