From 2fe22da3a234935ada2991ac1c35aed15fc2bd04 Mon Sep 17 00:00:00 2001 From: younesbelkada Date: Mon, 3 Apr 2023 16:47:09 +0000 Subject: [PATCH] fix CI --- tests/test_common_gpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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