transformers/tests/models/esm
Younes Belkada 50db7ca4e8
FIX [`quantization` / `ESM`] Fix ESM 8bit / 4bit with bitsandbytes (#29329)
* 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>
2024-03-01 03:01:53 +01:00
..
__init__.py Rebase ESM PR and update all file formats (#19055) 2022-09-30 14:16:25 +01:00
test_modeling_esm.py FIX [`quantization` / `ESM`] Fix ESM 8bit / 4bit with bitsandbytes (#29329) 2024-03-01 03:01:53 +01:00
test_modeling_esmfold.py CI with `num_hidden_layers=2` 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
test_modeling_tf_esm.py Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
test_tokenization_esm.py Allow add_tokens for ESM (#28535) 2024-01-19 12:32:05 +00:00