tensor return error fix

neuralmagic · Apr 29, 2024 · d3f0803 · d3f0803
1 parent 8a2224f
commit d3f0803
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/compressed_tensors/quantization/observers/base.py b/src/compressed_tensors/quantization/observers/base.py
@@ -124,4 +124,4 @@ def get_qparams_per_channel(self, observed, channel: int):
             scales.append(scale)
             zero_points.append(zero_point)
 
-        return torch.cat(scales), torch.cat(zero_points)
+        return torch.stack(scales), torch.stack(zero_points)
diff --git a/src/compressed_tensors/quantization/observers/min_max.py b/src/compressed_tensors/quantization/observers/min_max.py
@@ -40,7 +40,6 @@ def __init__(
         self.max_val = -float("inf")
         self.averaging_constant = averaging_constant
 
-
     def calculate_qparams(self, observed: Tensor) -> Tuple[FloatTensor, IntTensor]:
         """
         Updates the observed min and max using a moving average smoothed by the