diffusers
[Core] add support for gradient checkpointing in transformer_2d
#5943
Merged

[Core] add support for gradient checkpointing in transformer_2d #5943

sayakpaul merged 2 commits into main from feat/grad-ckpt-transformer
sayakpaul
sayakpaul add support for gradient checkpointing in transformer_2d
d457826c
sayakpaul sayakpaul requested a review from patrickvonplaten patrickvonplaten 2 years ago
HuggingFaceDocBuilderDev
patrickvonplaten
patrickvonplaten approved these changes on 2023-11-27
sayakpaul Merge branch 'main' into feat/grad-ckpt-transformer
edb7aa77
sayakpaul sayakpaul merged 3f7c3511 into main 2 years ago
sayakpaul sayakpaul deleted the feat/grad-ckpt-transformer branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone