axis=1 check

mobiusml · May 6, 2024 · 108cdf6 · 108cdf6
1 parent f5a2f7e
commit 108cdf6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hqq/backends/torchao.py b/hqq/backends/torchao.py
@@ -68,7 +68,7 @@ def __init__(
         self.inner_k_tiles = inner_k_tiles
         self.padding = padding
 
-        assert self.axis, "Only axis==1 is supported"
+        assert self.axis==1, "Only axis==1 is supported"
         assert self.nbits in [4], "Unsupported nbits."
         assert (
             self.compute_dtype is bfloat16