diffusers

mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Files

dg845 648d968cfc Enable Gradient Checkpointing for UNet2DModel (New) (#7201 )

* Port UNet2DModel gradient checkpointing code from #6718.


---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Co-authored-by: Vincent Neemie <92559302+VincentNeemie@users.noreply.github.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
Co-authored-by: hlky <hlky@hlky.ac>

2024-12-19 14:45:45 -10:00

__init__.py

[Refactor] harmonize the module structure for models in tests (#6738 )

2024-02-01 14:23:39 +05:30

test_models_unet_1d.py

[Tests] reduce block sizes of UNet and VAE tests (#7560 )

2024-04-05 10:08:32 +05:30

test_models_unet_2d_condition.py

[LoRA] depcrecate save_attn_procs(). (#10126 )

2024-12-06 10:38:57 -10:00

test_models_unet_2d_flax.py

[Refactor] harmonize the module structure for models in tests (#6738 )