src/diffusers/models/attention_processor.py · 87252d80c3ea8eb6fba8b6de8c2dac9ede4fadee · renzhc / diffusers_dcu

"...resnet50_tensorflow.git" did not exist on "fe7202ee66a8adffb08022b366ae3831b81eec14"

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

attention_processor.py 256 KB

Replace attention_processor.py