guided-diffusion
guided-diffusion copied to clipboard
Adjusting `cond_fn` for enabling `nn.DataParallel` to use Multi-GPUs
Hello there,
I hope you all doing well. Sorry If my question looks silly, but I want to know If there is a capability of adding Diffusion into nn.DataParallel for paralyzing the generating condition among multiple GPUs.
Or If there is a possibility to adjust guided-diffusion diffusion to run amount multiple-GPUs.
regards,