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