Files
HuggingFace_transformer/tests
Younes Belkada f1e8c48c5e Add accelerate support for ViT family (#20174)
* 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
2022-11-15 11:06:01 +01:00
..
2022-11-14 01:04:26 -05:00
2022-11-14 01:00:10 -05:00
2022-10-18 13:47:36 -04:00
2022-09-15 11:10:16 -07:00
2022-11-02 11:57:36 +00:00