NadaGh's picture
End of training
3a25a0a verified

DiTTransformer2DModel

A Transformer model for image-like data from DiT.

DiTTransformer2DModel

[[autodoc]] DiTTransformer2DModel