Duplicated from JunhaoZhuang/Cobra
A Transformer model for image-like data from DiT.
[[autodoc]] DiTTransformer2DModel