diff --git a/src/compressed_tensors/quantization/quant_scheme.py b/src/compressed_tensors/quantization/quant_scheme.py index a9c8b45a2..fedd84e5a 100644 --- a/src/compressed_tensors/quantization/quant_scheme.py +++ b/src/compressed_tensors/quantization/quant_scheme.py @@ -60,6 +60,16 @@ def validate_model_after(model: "QuantizationScheme") -> "QuantizationScheme": format = model.format if inputs is not None: + if inputs.strategy not in ( + QuantizationStrategy.TOKEN, + QuantizationStrategy.TENSOR, + QuantizationStrategy.TENSOR_GROUP, + ): + raise ValueError( + f"Using {inputs.strategy} strategy is not supported for " + "activation quantization" + ) + if inputs.actorder is not None: raise ValueError("Cannot apply actorder to input activations")