src/diffusers/models/unet_2d_blocks.py · 2858d7e15eaf445ec37fc77b204a85f84affbeef · renzhc / diffusers_dcu

Fix gradient checkpointing bugs in freezing part of models (requires_grad=False) (#3404) · 15f1bab1

7eu7d7 authored May 17, 2023



* gradient checkpointing bug fix

* bug fix; changes for reviews

* reformat

* reformat

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

15f1bab1

unet_2d_blocks.py 106 KB

Replace unet_2d_blocks.py