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

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

まず DiT モデルと PixArt モデルのエントリポイントを特定し、transformer モデルが output_attentions をどのように処理しているかを比較します。これらのモデルで期待される attention 出力の形状と動作を確認し、その後、要求されたオプションのカバレッジを追加して、そのオプションを有効にするとデフォルトの動作を変更せずに attention 出力が返されることを検証します。

索引モデルが issue の本文から書いたものです。