transformers
Switch gradient checkpointing default to `use_reentrant=False` (PyTorch recommended)
#43203
Merged
Overview
Commits
33
Changes
View On GitHub

Switch gradient checkpointing default to `use_reentrant=False` (PyTorch recommended) #43203

ArthurZucker merged 33 commits into main from no-reentrant
qgallouedec
qgallouedec Switch gradient checkpointing default to use_reentrant=False (PyTorch…
7e5c7d1a
qgallouedec qgallouedec changed the title Switch gradient checkpointing default to use_reentrant=False (PyTorch… Switch gradient checkpointing default to use_reentrant=False (PyTorch recommended) 200 days ago
qgallouedec qgallouedec changed the title Switch gradient checkpointing default to use_reentrant=False (PyTorch recommended) Switch gradient checkpointing default to `use_reentrant=False` (PyTorch recommended) 200 days ago
HuggingFaceDocBuilderDev
qgallouedec All testers
7ea2cbe0
qgallouedec revert
340224cf
qgallouedec Merge branch 'main' into no-reentrant
68c0711c
qgallouedec skip for the right reason
d642e529
qgallouedec skip for right reason and this one is fixed
1210ec2b
qgallouedec skip for the right reason
453d5cc9
qgallouedec fix up to blip2 excluded
4356551d
qgallouedec
qgallouedec commented on 2026-01-09
qgallouedec reactivate all tests
4ac2ceeb
qgallouedec qgallouedec changed the title Switch gradient checkpointing default to `use_reentrant=False` (PyTorch recommended) [WIP] Switch gradient checkpointing default to `use_reentrant=False` (PyTorch recommended) 200 days ago
qgallouedec skip non trainable and xfail know issues
71df8d66
qgallouedec audio encoder after lm initialization and add xfail tests for gradien…
d4154711
qgallouedec continue fixing tests
09a01fca
qgallouedec some last cases
aa7ed85b
qgallouedec Merge branch 'main' into no-reentrant
1e2c55c2
qgallouedec Merge branch 'no-reentrant' of https://github.com/huggingface/transfo…
c2688be0
qgallouedec fix check_training_gradient_checkpointing
525c1f39
qgallouedec ignore and x fail other tests
d2e1b8f5
qgallouedec this module doesn't support training
20ec2198
qgallouedec forgot one
3d090e14
qgallouedec mra
0dbb70f0
qgallouedec style
e0535d83
qgallouedec fix recurrent gemma when enable input require grads with gradient che…
314f63c3
qgallouedec mark training test as expected failure for reentrant compatibility
da4c49bc
qgallouedec style
a4603d52
qgallouedec reverting, moshi won't work anyway
84441417
qgallouedec
qgallouedec commented on 2026-01-09
qgallouedec
qgallouedec commented on 2026-01-11
qgallouedec qgallouedec changed the title [WIP] Switch gradient checkpointing default to `use_reentrant=False` (PyTorch recommended) Switch gradient checkpointing default to `use_reentrant=False` (PyTorch recommended) 198 days ago
qgallouedec remove skip training tests for Siglip2VisionModel
09012a43
qgallouedec qgallouedec requested a review from ydshieh ydshieh 198 days ago
qgallouedec qgallouedec requested a review from SunMarc SunMarc 198 days ago
qgallouedec qgallouedec requested a review from ArthurZucker ArthurZucker 198 days ago
qgallouedec qgallouedec requested a review from Cyrilvallez Cyrilvallez 198 days ago
qgallouedec Merge branch 'main' into no-reentrant
8c09f171
qgallouedec Merge branch 'main' into no-reentrant
183d3a87
SunMarc
SunMarc approved these changes on 2026-01-12
SunMarc Merge branch 'main' into no-reentrant
36cb28dc
SunMarc Merge branch 'main' into no-reentrant
c7bf3e25
SunMarc Merge branch 'main' into no-reentrant
a9c9a66d
SunMarc Merge branch 'main' into no-reentrant
1926ac8d
SunMarc Merge branch 'main' into no-reentrant
75655a8a
github-actions
github-actions
ArthurZucker ArthurZucker merged 625ad844 into main 194 days ago
ArthurZucker ArthurZucker deleted the no-reentrant branch 194 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum