riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 11:11:05 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity
d21c97cc0f
transformers/tests/models/deit
History
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_deit.py Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
test_modeling_deit.py Add accelerate support for ViT family (#20174) 2022-11-15 11:06:01 +01:00
test_modeling_tf_deit.py Add TF DeiT implementation (#17806) 2022-07-13 18:04:08 +01:00