src/diffusers/models/attention_flax.py · a5f35ee4731b731d6bd8977525873b0bc480cb42 · renzhc / diffusers_dcu

Juan Acevedo authored Dec 14, 2024


Co-authored-by: Juan Acevedo <jfacevedo@google.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Co-authored-by: Aryan <aryan@huggingface.co>

a5f35ee4

attention_flax.py 19.8 KB