fix merge error (#325)

intel · Nov 15, 2024 · dadb95a · dadb95a
1 parent d493572
commit dadb95a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/auto_round/backend.py b/auto_round/backend.py
@@ -404,7 +404,7 @@ def get_autogptq_infer_linear(backend, bits=4, group_size=128, sym=False):
     from packaging.version import Version
 
     # Import the appropriate QuantLinear based on the version of auto_gptq
-    if Version(version) <= Version("0.7.1"):
+    if Version(version) < Version("0.7.2"):
         QuantLinear = dynamically_import_QuantLinear(
             use_triton=use_triton,
             desc_act=False,