diffusers

[Core] add support for gradient checkpointing in transformer_2d

#5943

Merged