Introduce GradientCheckpointingLayer #37223
GradientCheckpointingLayer
486f1553
qubvel
marked this pull request as ready for review 364 days ago
trigger
e5d326a4
Move GC layer to a separate file
70dc32e7
Update import
9f8c1ce4
Expose and document GC layer
fc96fadb
Merge branch 'main' into gradient-checkpointing-layer
657c538a
Merge branch 'main' into gradient-checkpointing-layer
5a7dd6b8
Fix dummy
72baa13c
Apply to llama-based models
31f67201
Update modulars
334043a3
Update a few more models for consistency
d43dfd43
Merge branch 'main' into gradient-checkpointing-layer
da0de606
Update glm4
23e6e24e
Merge branch 'main' into gradient-checkpointing-layer
2093264a
Merge branch 'main' into gradient-checkpointing-layer
2e913fd9
Update Janus
4640e46b
qubvel
merged
9167fada
into main 344 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub