当前位置:   article > 正文

ChatGLM推理报错RuntimeError: probability tensor contains either `inf`, `nan` or element < 0

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

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

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

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/57761
推荐阅读
相关标签
  

闽ICP备14008679号