Adjusting `cond_fn` for enabling `nn.DataParallel` to use Multi-GPUs

openai / guided-diffusion

MIT License

6.06k stars 807 forks source link

Adjusting `cond_fn` for enabling `nn.DataParallel` to use Multi-GPUs #45

Open AI-Ahmed opened 2 years ago

AI-Ahmed commented 2 years ago

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,