• Younes Belkada's avatar
    Add `accelerate` support for `ViT` family (#20174) · f1e8c48c
    Younes Belkada authored
    * add `accelerate` support for `ViT` family
    
    - add `_no_split_modules`
    - manually cast to the right `dtype`: to change
    
    * enable `float16` for `deit`
    
    * fix `make fixup`
    
    * add `slow` test for `fp16` inference
    
    * another safety check
    
    * Update src/transformers/models/deit/modeling_deit.py
    f1e8c48c
test_modeling_deit.py 15.7 KB