Scalable Diffusion Models with Transformers (NOT OFFICIAL)

This repo is not accepting pull requests! Sorry!

Current code state: Fixed the training script (UNTESTED)

Author(s): William Peebles and Saining Xie

Paper: https://arxiv.org/abs/2212.09748

Official code: https://github.com/facebookresearch/DiT

What makes this different?

• Complete code rewrite from scratch
• Modularized parts for better experimentation
• Added expected tensor shape guides
• Upgraded training script