HelloVision / HelloMeme

The official HelloMeme GitHub site
https://songkey.github.io/hellomeme/
MIT License
216 stars 14 forks source link

new error - found at least two devices, cpu and cuda: #13

Open lior007 opened 2 weeks ago

lior007 commented 2 weeks 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 mat1 in method wrapper_CUDA_addmm)

songkey commented 2 weeks ago

I noticed this issue as well—it happens occasionally, though the root cause hasn't been identified yet. For now, I’ve temporarily added a device parameter to reduce its occurrence, and I plan to investigate it thoroughly in the future.

https://github.com/HelloVision/HelloMeme/blob/aefce6ef65aca00ab1cb58f884df64479e7d0698/inference_image.py#L108