src/diffusers/models/transformer_2d.py · bf40d7d82a732a35e0f5b907e59064ee080cde9f · renzhc / diffusers_dcu · GitLab

"docker/Dockerfile.ci_gpu_cu11" did not exist on "92321347c06e43b9e6e35add8a5caec7dbf27b3f"

Find file Blame History Permalink

[Core] add support for gradient checkpointing in transformer_2d (#5943) · 3f7c3511
Sayak Paul authored Nov 27, 2023
```
add support for gradient checkpointing in transformer_2d
```
3f7c3511

transformer_2d.py 23.2 KB