Add cross attention to DiT blocks#34

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

test_888717187google/hackable_diffusion:test_888717187Copy head branch name to clipboard