Skip to content

Commit 2922841

Browse files
committedFeb 8, 2024
Applied more comments
1 parent c0e5a1a commit 2922841

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed
 

‎optimum/intel/openvino/quantization.py

+4
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,10 @@ def quantize(
223223
"`calibration_dataset` is needed to compute the activations range during the calibration step and was not provided. "
224224
"In case you only want to apply quantization on the weights, please set `weights_only=True`."
225225
)
226+
quantization_config = kwargs.pop("quantization_config", None)
227+
if quantization_config is not None:
228+
logger.warning("The argument `quantization_config` is deprecated, and will be removed in optimum-intel v1.6.0, please use `ov_config` instead")
229+
ov_config = ov_config or quantization_config
226230

227231
if isinstance(self.model, OVBaseDecoderModel) and self.model.use_cache:
228232
self._quantize_ovcausallm(

0 commit comments

Comments
 (0)