transformers
feat: add support for gradient checkpointing for TimmWrapperModel and TimmWrapperForImageClassification
#39287

Merged

feat: add support for gradient checkpointing for TimmWrapperModel and TimmWrapperForImageClassification #39287

qubvel merged 9 commits into huggingface:main from Yozer:gradient-checkpoint-timm

feat: add support for gradient checkpointing in TimmWrapperModel and …

f60a8353

ruff fix

2b75a68d

qubvel commented on 2025-07-15

refactor + add test for not supported model

31a518c4

ruff

c8270a88

qubvel commented on 2025-07-16

Update src/transformers/models/timm_wrapper/modeling_timm_wrapper.py

fa5bf187

Update src/transformers/models/timm_wrapper/modeling_timm_wrapper.py

25517823

Update src/transformers/models/timm_wrapper/modeling_timm_wrapper.py

305ffe3b

Update src/transformers/models/timm_wrapper/modeling_timm_wrapper.py

41136fce

qubvel approved these changes on 2025-07-22

Merge branch 'main' into gradient-checkpoint-timm

6d79e874

qubvel enabled auto-merge (squash) 177 days ago

qubvel merged 30567c28 into main 177 days ago

Reviewers

qubvel

Assignees

No one assigned

Labels

None yet

Milestone

No milestone