DiT: How Transformers Are Replacing U-Nets in Diffusion Models

Replacing U-Nets with Vision Transformers for high-quality, compute-scalable image generation