Is there any DiT block defined in the huggingface/diffusers OR huggingface/transformers project?

[JohnHerry]

Is your feature request related to a problem? Please describe.
I want to make some experiments about DiT based flow-matching model, I need an implementation of the common DiT block, but did not found it in both huggingface/diffusers and huggingface/transformers. Is there any implementation about it with just some other file names?

Describe the solution you'd like.
A clear DiT implementation

Describe alternatives you've considered.

Additional context.

[a-r-r-o-w]

All the blocks implemented in this folder are blocks from a DiT: https://github.com/huggingface/diffusers/tree/main/src/diffusers/models/transformers

[JohnHerry]

Thank you for the help, it seems like the dit_transformer_2d.py somewhat relative, but without support for sequence_padding_mask, which let it can not be used for training.