diff --git a/tests/test_common_gpu.py b/tests/test_common_gpu.py index 2f30018..d9099ce 100644 --- a/tests/test_common_gpu.py +++ b/tests/test_common_gpu.py @@ -20,7 +20,7 @@ import torch from transformers import AutoModelForCausalLM, AutoModelForSeq2SeqLM, AutoTokenizer, WhisperForConditionalGeneration from peft import LoraConfig, PeftModel, get_peft_model -from peft.utils import is_bnb_available +from peft.import_utils import is_bnb_available from .testing_utils import require_bitsandbytes, require_torch_gpu, require_torch_multi_gpu