diffusion transformers

Outlier Tokens in DiTs, and How DSR Fixes Them

A new paper shows outlier tokens affect both RAE encoders and DiT denoisers, and proposes Dual-Stage Registers to reduce artifacts.