diff --git a/olive/cli/auto_opt.py b/olive/cli/auto_opt.py index 3d75469b0..5303da1b7 100644 --- a/olive/cli/auto_opt.py +++ b/olive/cli/auto_opt.py @@ -443,7 +443,7 @@ def _get_passes_config(self, config: Dict[str, Any], olive_config: OlivePackageC ("qnn_preprocess", {"type": "QNNPreprocess"}), ("mixed_precision_overrides", {"type": "MixedPrecisionOverrides", "overrides_config": None}), # quantization passes - ("dynamic_quant", {"type": "OnnxQuantization", "weight_type": "QInt8"}), + ("dynamic_quant", {"type": "OnnxDynamicQuantization", "weight_type": "QInt8"}), ("matmul4", {"type": "OnnxMatMul4Quantizer"}), ("bnb4", {"type": "OnnxBnb4Quantization", "quant_type": "nf4"}), # post processing passes