stablediffusion
stablediffusion copied to clipboard
DDP mode problem
Hi ,thanks for your great work!
when i use DDP mode to run the code, it shows the problem: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!
could you help me to solve it? thanks !