src/diffusers/models/modeling_utils.py · aeac0a00f88dccce233c062f27d59028ed195d9f · renzhc / diffusers_dcu

"vscode:/vscode.git/clone" did not exist on "9efe1e52c3f20ed02ad91097b3e471a45e1ba8ed"

implementing flux on TPUs with ptxla (#10515) · aeac0a00

Juan Acevedo authored Jan 16, 2025



* implementing flux on TPUs with ptxla

* add xla flux attention class

* run make style/quality

* Update src/diffusers/models/attention_processor.py
Co-authored-by: YiYi Xu <yixu310@gmail.com>

* Update src/diffusers/models/attention_processor.py
Co-authored-by: YiYi Xu <yixu310@gmail.com>

* run style and quality

---------
Co-authored-by: Juan Acevedo <jfacevedo@google.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Co-authored-by: YiYi Xu <yixu310@gmail.com>

aeac0a00

modeling_utils.py 70 KB

Replace modeling_utils.py