tests/bnb/test_mixed_int8.py · 12240925cfa29fff932e49927eb9744713ab1018 · chenpangpang / transformers

Add bitsandbytes support for gpt2 models (#24504) · 12240925

Dario Sučić authored Jun 28, 2023



* Add bitsandbytes support for gpt2 models

* Guard Conv1D import to pass tensorflow test

* Appease ruff linter

* Fix 4bit test and remove int8 test boilerplate

* Update tests/bnb/test_mixed_int8.py
Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>

---------
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>

12240925

test_mixed_int8.py 30 KB

Replace test_mixed_int8.py