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

..

__init__.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_feature_extraction_vit.py

Add Image Processors (#19796)

2022-11-02 11:57:36 +00:00

test_modeling_flax_vit.py

Improve vision models (#17731)

2022-06-24 11:34:51 +02:00

test_modeling_tf_vit.py

Improve vision models (#17731)

2022-06-24 11:34:51 +02:00

test_modeling_vit.py

Add accelerate support for ViT family (#20174)

2022-11-15 11:06:01 +01:00