Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Samples #40

Open
wants to merge 8 commits into
base: inference
Choose a base branch
from
Prev Previous commit
Next Next commit
Don't pin bits and bytes, this old version won't cache 4-bit conversions
Error: You are calling `save_pretrained` to a 4-bit converted model, but your `bitsandbytes` version doesn't support it. If you want to save 4-bit models, make sure to have `bitsandbytes>=0.41.3` installed.

Signed-off-by: Alastair D'Silva <[email protected]>
  • Loading branch information
deece committed May 26, 2024
commit b71b9bfb2a98f77e757328e010da3e567e7440db
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ train = [
"accelerate>=0.29.1",
"tokenizers~=0.15.2",
"transformers@git+https://github.com/huggingface/transformers.git@1c39974a4c4036fd641bc1191cc32799f85715a4",
"bitsandbytes==0.41.0",
"bitsandbytes>=0.41.3",
"scikit-learn==1.2.2",
"sentencepiece~=0.1.99",
"einops==0.6.1",
Expand Down