transformers
fix: prevent accelerate from splitting vision encoder by setting _no_…
#43047
Merged

fix: prevent accelerate from splitting vision encoder by setting _no_… #43047

CodersAcademy006
ArthurZucker
ArthurZucker approved these changes on 2026-01-05
ArthurZucker
fix: add TimmWrapperForImageClassification to _no_split_modules in mo…
acd95861
CodersAcademy006
fix: add TimmWrapperForImageClassification to _no_split_modules in pe…
2034a812
ArthurZucker Merge branch 'main' into main
36d76e9e
ArthurZucker ArthurZucker enabled auto-merge 65 days ago
github-actions
HuggingFaceDocBuilderDev
ArthurZucker ArthurZucker merged 27fbb514 into main 64 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone