DB-GPT
DB-GPT copied to clipboard
Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat2 in method wrapper_CUDA_addmm
Search before asking
- [X] I had searched in the issues and found no similar issues.
Operating system information
Linux
Python version information
=3.11
DB-GPT version
main
Related scenes
- [X] Chat Data
- [X] Chat Excel
- [X] Chat DB
- [X] Chat Knowledge
- [X] Model Management
- [ ] Dashboard
- [ ] Plugins
Installation Information
-
[ ] AutoDL Image
-
[ ] Other
Device information
GPU Count:1 GPU M:24G
Models information
LLM:chatGLM3 6b
What happened
Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat2 in method wrapper_CUDA_addmm
What you expected to happen
input "hello" and press enter Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat2 in method wrapper_CUDA_addmm
How to reproduce
Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat2 in method wrapper_CUDA_addmm
Additional context
No response
Are you willing to submit PR?
- [ ] Yes I am willing to submit a PR!
found same Issue.
This issue has been marked as stale, because it has been over 30 days without any activity.
This issue bas been closed, because it has been marked as stale and there has been no activity for over 7 days.