tests/deepspeed/test_deepspeed.py · 7ecd229ba475dbf78040f368ae86c86bba875442 · chenpangpang / transformers

Smangrul/fix failing ds ci tests (#27358) · 7ecd229b

Sourab Mangrulkar authored Nov 09, 2023

* fix failing DeepSpeed CI tests due to `safetensors` being default

* debug

* remove debug statements

* resolve comments

* Update test_deepspeed.py

7ecd229b

test_deepspeed.py 47.9 KB

Replace test_deepspeed.py