transformers/tests/deepspeed
Fanli Lin 9454f437b0
[tests] make TestDeepSpeedModelZoo device-agnostic (#31402)
* fix

* use accelerator device count

* ci fix
2024-06-17 16:42:57 +02:00
..
ds_config_zero2.json [Deepspeed] add support for bf16 mode (#14569) 2022-03-11 17:53:53 -08:00
ds_config_zero3.json Update ds_config_zero3.json (#30829) 2024-05-15 10:02:31 -04:00
test_deepspeed.py 🚨🚨🚨Deprecate evaluation_strategy to eval_strategy🚨🚨🚨 (#30190) 2024-04-18 12:49:43 -04:00
test_model_zoo.py [tests] make TestDeepSpeedModelZoo device-agnostic (#31402) 2024-06-17 16:42:57 +02:00
vit_feature_extractor.json missing file (#17164) 2022-05-10 10:19:50 -07:00