src/diffusers/models/attention.py · daebee0963d2b39fb3fa9532ab271a91674c4070 · renzhc / diffusers_dcu

Add xformers attention to VAE (#1507) · daebee09

Ilmari Heikkinen authored Dec 03, 2022



* Add xformers attention to VAE

* Simplify VAE xformers code

* Update src/diffusers/models/attention.py
Co-authored-by: Ilmari Heikkinen <ilmari@fhtr.org>
Co-authored-by: Suraj Patil <surajp815@gmail.com>

daebee09

attention.py 37.3 KB

Replace attention.py