Tensor Group Validation (#490)

kylesayrs · web-flow · commit 1ef272366577 · 2025-10-16T10:59:15.000-04:00
Signed-off-by: Kyle Sayers &lt;kylesayrs@gmail.com&gt;
diff --git a/src/compressed_tensors/quantization/quant_args.py b/src/compressed_tensors/quantization/quant_args.py
@@ -287,7 +287,7 @@ def validate_model_after(model: "QuantizationArgs") -> "QuantizationArgs":
             )
 
         # validate group strategy
-        if strategy == QuantizationStrategy.GROUP:
+        if strategy in (QuantizationStrategy.GROUP, QuantizationStrategy.TENSOR_GROUP):
             if group_size is None or group_size <= 0:
                 raise ValueError(
                     f"strategy {strategy} requires group_size to be "

Original file line number	Diff line number	Diff line change
`@@ -287,7 +287,7 @@ def validate_model_after(model: "QuantizationArgs") -> "QuantizationArgs":`
`287`	`287`	`)`
`288`	`288`
`289`	`289`	`# validate group strategy`
`290`		`- if strategy == QuantizationStrategy.GROUP:`
	`290`	`+ if strategy in (QuantizationStrategy.GROUP, QuantizationStrategy.TENSOR_GROUP):`
`291`	`291`	`if group_size is None or group_size <= 0:`
`292`	`292`	`raise ValueError(`
`293`	`293`	`f"strategy {strategy} requires group_size to be "`