ChatGLM推理报错RuntimeError: probability tensor contains either `inf`, `nan` or element ＜ 0
ChatGLM推理报错RuntimeError: probability tensor contains either `inf`, `nan` or element < 0
·
训练完ChatGLM后想执行推理时，发现模型报错
RuntimeError: probability tensor contains either `inf`, `nan` or element < 0
解决方法：
删除ptuning下的main.py文件第387行和396行的do_sample=True红框语句。
一站式 AI 云服务平台
更多推荐
3
0- 0
已为社区贡献3条内容
所有评论(0)