simplify activation shape

kylesayrs · kylesayrs · commit 72560d4dfbe4 · 2025-09-30T17:27:27.000-04:00
Signed-off-by: Kyle Sayers &lt;kylesayrs@gmail.com&gt;
diff --git a/src/compressed_tensors/quantization/lifecycle/initialize.py b/src/compressed_tensors/quantization/lifecycle/initialize.py
@@ -105,7 +105,7 @@ def initialize_module_for_quantization(
                 module,
                 "input",
                 scheme.input_activations,
-                observed_shape=(1, weight.shape[-1]),
+                observed_shape=weight.shape[-1:],
                 observed_dtype=weight.dtype,
                 force_zero_point=force_zero_point,
             )

Original file line number	Diff line number	Diff line change
`@@ -105,7 +105,7 @@ def initialize_module_for_quantization(`
`105`	`105`	`module,`
`106`	`106`	`"input",`
`107`	`107`	`scheme.input_activations,`
`108`		`- observed_shape=(1, weight.shape[-1]),`
	`108`	`+ observed_shape=weight.shape[-1:],`
`109`	`109`	`observed_dtype=weight.dtype,`
`110`	`110`	`force_zero_point=force_zero_point,`
`111`	`111`	`)`