🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
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 <[email protected]> --------- Co-authored-by: Sylvain Gugger <[email protected]> Co-authored-by: Younes Belkada <[email protected]>
D
Dario Sučić committed
12240925cfa29fff932e49927eb9744713ab1018
Parent: 89b6ee4
Committed by GitHub <[email protected]>
on 6/28/2023, 3:55:32 AM