Skip to content

Commit

Permalink
update min version of autogptq
Browse files Browse the repository at this point in the history
  • Loading branch information
SunMarc committed Oct 24, 2023
1 parent d2bd4af commit a8cefa7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions optimum/utils/import_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,11 +110,11 @@ def is_timm_available():
def is_auto_gptq_available():
if _auto_gptq_available:
version_autogptq = packaging.version.parse(importlib_metadata.version("auto_gptq"))
if AUTOGPTQ_MINIMUM_VERSION <= version_autogptq:
if AUTOGPTQ_MINIMUM_VERSION < version_autogptq:
return True
else:
raise ImportError(
f"Found an incompatible version of auto-gptq. Found version {version_autogptq}, but only {AUTOGPTQ_MINIMUM_VERSION} and above are supported"
f"Found an incompatible version of auto-gptq. Found version {version_autogptq}, but only version above {AUTOGPTQ_MINIMUM_VERSION} are supported"
)


Expand Down

0 comments on commit a8cefa7

Please sign in to comment.