tests/mixed_int8/test_mixed_int8.py · 7bbdfd7b240141dd578610e0bcae2fe62cc13451 · chenpangpang / transformers

Fix accelerate logger bug (#23650) · 7bbdfd7b

Younes Belkada authored May 22, 2023



* fix logger bug

* Update tests/mixed_int8/test_mixed_int8.py
Co-authored-by: Zachary Mueller <muellerzr@gmail.com>

* import `PartialState`

---------
Co-authored-by: Zachary Mueller <muellerzr@gmail.com>

7bbdfd7b

test_mixed_int8.py 27.2 KB

Replace test_mixed_int8.py