transformers
27fbb514 - fix: prevent accelerate from splitting vision encoder by setting _no_… (#43047)

Commit
28 days ago
fix: prevent accelerate from splitting vision encoder by setting _no_… (#43047) * fix: add TimmWrapperForImageClassification to _no_split_modules in modular files and regenerate - Update modular_pe_audio_video.py and modular_pe_video.py (source of truth) - Regenerate modeling_pe_audio_video.py and modeling_pe_video.py via modular_model_converter.py - Remove @unittest.skip on test_model_parallelism now that the crash is resolved Fixes #42918 * fix: add TimmWrapperForImageClassification to _no_split_modules in pe_audio, pe_video, pe_audio_video --------- Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
Parents
Loading