Skip to content

Commit 8204fd7

Browse files
committed
fix configuration
1 parent 80ad67a commit 8204fd7

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

optimum/intel/openvino/configuration.py

+5-7
Original file line numberDiff line numberDiff line change
@@ -118,14 +118,12 @@ def __init__(
118118
self.save_onnx_model = save_onnx_model
119119
self._enable_standard_onnx_export_option()
120120
self.optimum_version = kwargs.pop("optimum_version", None)
121-
self.quantization_config = quantization_config
122-
123-
bits = None
124-
if isinstance(quantization_config, dict):
125-
bits = quantization_config.get("bits", None)
126-
elif isinstance(quantization_config, QuantizationConfigMixin):
127-
bits = quantization_config.bits
121+
self.quantization_config = quantization_config or {}
128122

123+
if isinstance(quantization_config, QuantizationConfigMixin):
124+
bits = self.quantization_config.bits
125+
else:
126+
bits = self.quantization_config.get("bits", None)
129127
self.dtype = "int" + str(bits) if isinstance(bits, int) else dtype
130128

131129
def add_input_info(self, model_inputs: Dict, force_batch_one: bool = False):

0 commit comments

Comments
 (0)