transformers/tests/deepspeed
Zach Mueller 82efc53513
Yell at the user if zero-3 init wasn't performed, but expected to have been done (#32299)
* Test this zach

* Test for improper init w/o zero3

* Move back

* Apply suggestions from code review

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Get rid of stars in warning

* Make private

* Make clear

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2024-08-01 15:18:43 -04: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 Yell at the user if zero-3 init wasn't performed, but expected to have been done (#32299) 2024-08-01 15:18:43 -04:00
test_model_zoo.py Pass datasets trust_remote_code (#31406) 2024-06-17 17:29:13 +01:00
vit_feature_extractor.json missing file (#17164) 2022-05-10 10:19:50 -07:00