PR #24504 Add bitsandbytes support for gpt2 models

Add bitsandbytes support for gpt2 models #24504

younesbelkada merged 5 commits into huggingface:main from DarioSucic:main

Add bitsandbytes support for gpt2 models

1e64acbd

Guard Conv1D import to pass tensorflow test

1d5679ac

Appease ruff linter

82399a0f

younesbelkada commented on 2023-06-27

Fix 4bit test and remove int8 test boilerplate

b0ffa627

DarioSucic requested a review from

younesbelkada 2 years ago

younesbelkada approved these changes on 2023-06-27

younesbelkada requested a review from

sgugger 2 years ago

Update tests/bnb/test_mixed_int8.py

16dc7c4d

sgugger approved these changes on 2023-06-28

younesbelkada merged 12240925 into main 2 years ago

Reviewers

sgugger

younesbelkada

Assignees

No one assigned

Labels

None yet

Milestone

No milestone