Skip to content

Commit 8fe43a0

Browse files
Fix tests
1 parent 89b3afc commit 8fe43a0

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

optimum/intel/openvino/configuration.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -717,7 +717,10 @@ def __init__(
717717
self.compression = kwargs.get(
718718
"compression", None
719719
) # A field for backward-compatability of training-time compression parameters
720-
self.dtype = dtype
720+
if self.quantization_config is not None:
721+
self.dtype = self.quantization_config.weight_format
722+
else:
723+
self.dtype = dtype
721724

722725
def add_input_info(self, model_inputs: Dict, force_batch_one: bool = False):
723726
self.input_info = [

0 commit comments

Comments
 (0)