search

PKU-YuanGroup / ChatLaw

ChatLaw：A Powerful LLM Tailored for Chinese Legal. 中文法律大模型
https://chatlaw.cloud/
GNU Affero General Public License v3.0
6.92k stars 544 forks source link

模型合并好了，跑示例代码报错了 #41

Open xjlv1113 opened 1 year ago

xjlv1113 commented 1 year ago

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper_CUDA__index_select)a'

nuaabuaa07 commented 1 year ago

我报的错误和你相似，区别是在发现，在两个显卡上，不允许。还没找到解决方法