fix group dim (#2308)

neuralmagic · May 29, 2024 · bea67b3 · bea67b3
1 parent 1fcec3f
commit bea67b3
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/src/sparseml/modifiers/quantization/gptq/utils/gptq_wrapper.py b/src/sparseml/modifiers/quantization/gptq/utils/gptq_wrapper.py
@@ -195,11 +195,6 @@ def fasterprune(
                                 quant_scheme.weights,
                             )
                         else:  # strategy == QuantizationStrategy.GROUP
-                            # TODO: for grouped quantization its always 3d but the last
-                            # dim is always 1. Can we just make it 2d instead and avoid?
-                            scale = scale[:, :, 0]
-                            zero_point = zero_point[:, :, 0]
-
                             # get the group index for the current column
                             column_idx = i1 + i
                             input_dim_group = (