Commit c22e683 1 parent c441dac commit c22e683 Copy full SHA for c22e683
File tree 1 file changed +10
-0
lines changed
1 file changed +10
-0
lines changed Original file line number Diff line number Diff line change @@ -1021,6 +1021,11 @@ def _weight_only_quantization(
1021
1021
if isinstance (config , dict ):
1022
1022
config = OVWeightQuantizationConfig .from_dict (quantization_config )
1023
1023
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
+
1024
1029
dataset = None
1025
1030
if calibration_dataset is not None :
1026
1031
if is_datasets_available () and isinstance (calibration_dataset , Dataset ):
@@ -1054,6 +1059,11 @@ def _full_quantization(
1054
1059
verify_not_optimized : bool = True ,
1055
1060
** kwargs ,
1056
1061
):
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
+
1057
1067
if verify_not_optimized :
1058
1068
_verify_not_optimized (model )
1059
1069
q_kwargs = copy .deepcopy (kwargs )
You can’t perform that action at this time.
0 commit comments