Skip to content

Commit 94d9796

Browse files
committed
Get compression from kwargs
1 parent 42d8a8a commit 94d9796

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

optimum/intel/openvino/configuration.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,9 @@ def __init__(
310310
if isinstance(quantization_config, dict):
311311
quantization_config = self._quantization_config_from_dict(quantization_config)
312312
self.quantization_config = quantization_config
313-
self.compression = None # A field for backward-compatability of training-time compression parameters
313+
self.compression = kwargs.get(
314+
"compression", None
315+
) # A field for backward-compatability of training-time compression parameters
314316
bits = self.quantization_config.bits if self.quantization_config else None
315317
self.dtype = "int" + str(bits) if isinstance(bits, int) else dtype
316318

0 commit comments

Comments
 (0)