Can we pass output_attentions=True to DiT model such as pixart to get attention output?

[foreverpiano]

Can we pass output_attentions=True to DiT model such as pixart to get attention output? Like using output_attentions=True in transformer?

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.