Skip to content

Commit

Permalink
fix bnb version requirement
Browse files Browse the repository at this point in the history
  • Loading branch information
hiyouga committed Apr 26, 2023
1 parent a108614 commit fec3ad7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/utils/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -176,7 +176,7 @@ def load_pretrained(
if finetuning_args.finetuning_type != "p_tuning" and model_args.quantization_bit is not None:
if model_args.quantization_bit != 8:
raise ValueError("Freeze and LoRA fine-tuning only accept 8-bit quantization.")
require_version("bitsandbytes>=0.38.0", "bitsandbytes library is required to use this feature.")
require_version("bitsandbytes>=0.37.0", "bitsandbytes library is required to use this feature.")
from bitsandbytes.cuda_setup.main import get_compute_capability, get_cuda_lib_handle, is_cublasLt_compatible
cuda = get_cuda_lib_handle()
cc = get_compute_capability(cuda)
Expand Down

0 comments on commit fec3ad7

Please sign in to comment.