transformers/tests/deepspeed
Sourab Mangrulkar b477327394
fix the deepspeed tests (#26021)
* fix the deepspeed tests

* resolve comment
2023-09-13 10:26:53 +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 the deepspeed tests (#26021) 2023-09-13 10:26:53 +05:30
test_model_zoo.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
vit_feature_extractor.json missing file (#17164) 2022-05-10 10:19:50 -07:00