specify GPU

[ddzhen]

I have 2 GPUs when I want to use cuda:1, I try to modify gradio_*.py , but throw error like this: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0 Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0

[lllyasviel]

Thanks for the issue. it seems multi-gpu supprts need modifications of codes. please see also official docs of pytorch