test_sample.py运行报错，貌似是device不一样，untimeError: 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_gather) - Githubissues

chaoyi-wu / Finetune_LLAMA

简单易懂的LLaMA微调指南。

354 stars 34 forks source link

test_sample.py运行报错，貌似是device不一样，untimeError: 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_gather) #3

Open WUHU-G opened 1 year ago

sun-ruijiang commented 9 months ago

是否已经解决这个问题了呢，我也遇到了相同的问题 @FastBCSD @chaoyi-wu @xiaoman-zhang

sun-ruijiang commented 8 months ago

我已经解决这个问题了，转移到一个就行了