tests/deepspeed/test_model_zoo.py · 9454f437b0ba87e29cae9706cee6be6bb88d28d6 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[tests] make `TestDeepSpeedModelZoo` device-agnostic (#31402) · 9454f437
Fanli Lin authored Jun 17, 2024
```
* fix

* use accelerator device count

* ci fix
```
9454f437

test_model_zoo.py 12.7 KB