Skip to content

Commit c22e683

Browse files
Add explicit error on the wrong type of config
1 parent c441dac commit c22e683

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

optimum/intel/openvino/quantization.py

+10
Original file line numberDiff line numberDiff line change
@@ -1021,6 +1021,11 @@ def _weight_only_quantization(
10211021
if isinstance(config, dict):
10221022
config = OVWeightQuantizationConfig.from_dict(quantization_config)
10231023

1024+
if not isinstance(config, OVWeightQuantizationConfig):
1025+
raise ValueError(
1026+
f"Expected quantization config to be an instance of `OVWeightQuantizationConfig`, but got {type(config)}."
1027+
)
1028+
10241029
dataset = None
10251030
if calibration_dataset is not None:
10261031
if is_datasets_available() and isinstance(calibration_dataset, Dataset):
@@ -1054,6 +1059,11 @@ def _full_quantization(
10541059
verify_not_optimized: bool = True,
10551060
**kwargs,
10561061
):
1062+
if not isinstance(quantization_config, OVQuantizationConfig):
1063+
raise ValueError(
1064+
f"Expected quantization config to be an instance of `OVQuantizationConfig`, but got {type(quantization_config)}."
1065+
)
1066+
10571067
if verify_not_optimized:
10581068
_verify_not_optimized(model)
10591069
q_kwargs = copy.deepcopy(kwargs)

0 commit comments

Comments
 (0)