src/diffusers/models/unet_2d_blocks.py · a816a87a09e511f7b5dc81ce2f991892e5f3fffc · renzhc / diffusers_dcu

[refactor] Making the xformers mem-efficient attention activation recursive (#1493) · a816a87a

Benjamin Lefaudeux authored Dec 02, 2022



* Moving the mem efficiient attention activation to the top + recursive

* black, too bad there's no pre-commit ?
Co-authored-by: Benjamin Lefaudeux <benjamin@photoroom.com>

a816a87a

unet_2d_blocks.py 58.5 KB

Replace unet_2d_blocks.py