cloud_diffusion icon indicating copy to clipboard operation
cloud_diffusion copied to clipboard

Published 20 hours ago • verified publisher icon tcapelle

Distributed / Parallel version

Open miquel-espinosa opened this issue 1 year ago • 1 comments

Hi! Thanks for this project. Is there any distributed or parallel version, to run the code in multiple GPUs? Thanks

miquel-espinosa avatar Dec 01 '23 19:12 miquel-espinosa

Not right now, but I would gladly review a PR that adds this feature to the MiniTrainer class. You could use Accelerate or just vanilla DDP.

tcapelle avatar Dec 01 '23 22:12 tcapelle