Author(s): William Peebles and Saining Xie
Paper: https://arxiv.org/abs/2212.09748
Official code: https://github.com/facebookresearch/DiT
- Complete code rewrite from scratch
- Modularized parts for better experimentation
- Added expected tensor shape guides