diff --git a/src/compressed_tensors/quantization/quant_scheme.py b/src/compressed_tensors/quantization/quant_scheme.py index fedd84e5a..b11e3c0c0 100644 --- a/src/compressed_tensors/quantization/quant_scheme.py +++ b/src/compressed_tensors/quantization/quant_scheme.py @@ -63,9 +63,19 @@ def validate_model_after(model: "QuantizationScheme") -> "QuantizationScheme": if inputs.strategy not in ( QuantizationStrategy.TOKEN, QuantizationStrategy.TENSOR, + QuantizationStrategy.GROUP, QuantizationStrategy.TENSOR_GROUP, ): - raise ValueError( + if ( + inputs.strategy == QuantizationStrategy.GROUP + and inputs.dynamic is True + ): + raise NotImplementedError( + "Static and local group-wise activation " + "quantization is not supported" + ) + + raise NotImplementedError( f"Using {inputs.strategy} strategy is not supported for " "activation quantization" )