fix version parsing (#985)

ModelCloud · Jan 1, 2025 · 4f18747 · 4f18747
1 parent 244f17b
commit 4f18747
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gptqmodel/models/loader.py b/gptqmodel/models/loader.py
@@ -81,7 +81,7 @@ def check_versions(model_class, requirements: List[str]):
     for req in requirements:
         pkg, operator, version_required = parse_requirement(req)
         try:
-            installed_version = Version(pkg)
+            installed_version = version(pkg)
             if not compare_versions(installed_version, version_required, operator):
                 raise ValueError(f"{model_class} requires version {req}, but current {pkg} version is {installed_version} ")
         except PackageNotFoundError: