Skip to content

Commit af36b5b

Browse files
authored
Fix import typo autogptq (#116)
Should fix #113
1 parent 871192f commit af36b5b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lighteval/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def is_bnb_available() -> bool:
178178

179179

180180
def is_autogptq_available() -> bool:
181-
return importlib.util.find_spec("auto-gptq") is not None
181+
return importlib.util.find_spec("auto_gptq") is not None
182182

183183

184184
NO_AUTOGPTQ_ERROR_MSG = "You are trying to load a model quantized with `auto-gptq`, which is not available in your local environement. Please install it using pip."

0 commit comments

Comments
 (0)