Spaces:
Running
on
Zero
Running
on
Zero
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)
#13
by
RTayloor
- opened
Hi
I am running it locally and this is the error message I get when I try to use joycaption:
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)
I am using only one GPU
You need to delete @spaces.GPU decorators and move models to gpu. But I must warn you this spaces requires 34+ gb vram