tests/deepspeed/test_model_zoo.py · 4e3490f79b40248c53ee54365a9662611e880892 · chenpangpang / transformers

Sourab Mangrulkar authored Apr 09, 2024

* fix sequence length errors

* fix label column name error for vit

* fix the lm_head embedding!=linear layer mismatches for Seq2Seq models

4e3490f7

test_model_zoo.py 12.6 KB