transformers/tests/deepspeed
Sourab Mangrulkar 4e3490f79b
Fix failing DeepSpeed model zoo tests (#30112)
* fix sequence length errors

* fix label column name error for vit

* fix the lm_head embedding!=linear layer mismatches for Seq2Seq models
2024-04-09 12:01:47 +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 Fix failing DeepSpeed model zoo tests (#30112) 2024-04-09 12:01:47 +05:30
vit_feature_extractor.json missing file (#17164) 2022-05-10 10:19:50 -07:00