Dario Sučić
|
12240925cf
|
Add bitsandbytes support for gpt2 models (#24504)
* 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>
|
2023-06-28 05:55:32 +02:00 |
|