A pipeline parallel training script for diffusion models.
https://github.com/tdrussell/diffusion-pipe
fork for my own personal usage, credit to tdrussell for the great code!
A pipeline parallel training script for diffusion models.
https://github.com/tdrussell/diffusion-pipe
fork for my own personal usage, credit to tdrussell for the great code!