We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 31a856d commit b859c2fCopy full SHA for b859c2f
1 file changed
src/pquant/core/torch/quantizer.py
@@ -91,7 +91,7 @@ def compute_data_dynamic_bits(self, x):
91
return self.calculate_bits_from_abs(abs_x)
92
93
def compute_weight_dynamic_bits(self, x):
94
- if self.granularity == "per_tensor":
+ if self.granularity == "per_tensor" or x.ndim == 1:
95
_, i, f = self.get_quantization_bits()
96
return i, f
97
if self.granularity == "per_channel":
0 commit comments