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

ethansmith2000 authored Aug 07, 2023



* grad checkpointing

* fix make fix-copies

* fix

---------
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

f4f85413

unet_2d_blocks.py 119 KB