From 360de4e0b1d3f2aa3b569e1bf20665b6928b452a Mon Sep 17 00:00:00 2001 From: John Smith Date: Tue, 28 Mar 2023 23:51:35 +0800 Subject: [PATCH] Update lora.py --- src/peft/tuners/lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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():