transformers/tests/deepspeed
Sourab Mangrulkar b262808656
fix failing trainer ds tests (#29057)
2024-02-16 17:18:45 +05:30
..
ds_config_zero2.json [Deepspeed] add support for bf16 mode (#14569) 2022-03-11 17:53:53 -08:00
ds_config_zero3.json [Deepspeed] add support for bf16 mode (#14569) 2022-03-11 17:53:53 -08:00
test_deepspeed.py fix failing trainer ds tests (#29057) 2024-02-16 17:18:45 +05:30
test_model_zoo.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
vit_feature_extractor.json missing file (#17164) 2022-05-10 10:19:50 -07:00