src/diffusers/models/attention_dispatch.py · 354d35adb02e943d79014e5713290a4551d3dd01 · renzhc / diffusers_dcu

[core] Refactor hub attn kernels (#12475) · ab71f3c8

Sayak Paul authored Nov 19, 2025



* refactor how attention kernels from hub are used.

* up

* refactor according to Dhruv's ideas.
Co-authored-by: Dhruv Nair <dhruv@huggingface.co>

* empty
Co-authored-by: Dhruv Nair <dhruv@huggingface.co>

* empty
Co-authored-by: Dhruv Nair <dhruv@huggingface.co>

* empty
Co-authored-by: dn6 <dhruv@huggingface.co>

* up

---------
Co-authored-by: Dhruv Nair <dhruv@huggingface.co>
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

ab71f3c8

attention_dispatch.py 71.7 KB

Replace attention_dispatch.py