diff --git a/src/peft/tuners/lora.py b/src/peft/tuners/lora.py index 2547dc5..bfc7d13 100644 --- a/src/peft/tuners/lora.py +++ b/src/peft/tuners/lora.py @@ -33,7 +33,7 @@ def is_bnb_available(): def is_gptq_available(): - return importlib.util.find_spec("quant_cuda") is not None + return importlib.util.find_spec("gptq_llama.quant_cuda") is not None if is_bnb_available():