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 931df60 commit 4033f31Copy full SHA for 4033f31
src/compressed_tensors/quantization/lifecycle/initialize.py
@@ -218,7 +218,7 @@ def initialize_qparams(
218
219
group_size = quantization_args.group_size
220
num_groups = strategy_cdiv(observed_shape[-1], group_size, strategy)
221
- expected_shape = (*observed_shape[:-1], num_groups)
+ expected_shape = (observed_shape[-2], num_groups)
222
223
# initialize activation ordering if applicable
224
if actorder == ActivationOrdering.GROUP:
0 commit comments