Exception: Failed to invoke function <function inference_i2vgen_entrance at 0x7f56eb98d5e0>, with indices should be either on cpu or on the same device as the indexed tensor (cpu)

[howardgriffin]

File "/ldm_ani/i2vgen-xl/tools/modules/diffusions/diffusion_ddim.py", line 13, in _i return tensor[t].view(shape).to(x) RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu) How to solve it?

[Steven-SWZhang]

We found that this issue may arise in different environments. It has been fixed, you can pull the latest code again.

[Steven-SWZhang]

Or you can change the code, refer to here

[zhanghongyong123456]

Or you can change the code, refer to here

Wow, your method is so simple, I actually passed in the gpu id to modify the device