tests/models/esm/test_modeling_esm.py · 50db7ca4e874e211dd18d9b9ee429f62ef7d7e8f · chenpangpang / transformers

FIX [`quantization` / `ESM`] Fix ESM 8bit / 4bit with bitsandbytes (#29329) · 50db7ca4

Younes Belkada authored Mar 01, 2024



* fix ESM 8bit

* Apply suggestions from code review
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* fixup

---------
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

50db7ca4

test_modeling_esm.py 13.6 KB

Replace test_modeling_esm.py