Add cross attention to DiT#36

copybara-service[bot] wants to merge 1 commit intomaingoogle/hackable_diffusion:mainfrom

test_888888559google/hackable_diffusion:test_888888559Copy head branch name to clipboard