/REPA

Official Pytorch Implementation of Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think

Primary LanguagePythonMIT LicenseMIT

Stargazers