transformers/tests/deepspeed
Zach Mueller 4d8259d245
Fix loading zero3 weights (#36455)
* Check if fixes

* Fix zero3 loading

* Quality

* Fix marc nit

* Add fast tests

* Migrate to integrations.deepspeed rather than modeling_utils

* Style
2025-03-03 15:05:58 +01: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 Fix loading zero3 weights (#36455) 2025-03-03 15:05:58 +01: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