Official PyTorch Implementation of "Scalable Diffusion Models with Transformers" - View it on GitHub