tests/quantization/bnb/test_mixed_int8.py · 5ce90f32122f98942e3b54f0c4fbc3b393b396aa · chenpangpang / transformers

Bnb test fix for different hardwares (#29066) · 5ce90f32

Titus authored Feb 19, 2024



* generated text on A10G

* generated text in CI

* Apply suggestions from code review

add explanatory comments
Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>

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

5ce90f32

test_mixed_int8.py 35 KB

Replace test_mixed_int8.py