examples/community/sd_text2img_k_diffusion.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

sd_text2img_k_diffusion.py 21.7 KB

Replace sd_text2img_k_diffusion.py