We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 89b3afc commit 8fe43a0Copy full SHA for 8fe43a0
optimum/intel/openvino/configuration.py
@@ -717,7 +717,10 @@ def __init__(
717
self.compression = kwargs.get(
718
"compression", None
719
) # A field for backward-compatability of training-time compression parameters
720
- self.dtype = dtype
+ if self.quantization_config is not None:
721
+ self.dtype = self.quantization_config.weight_format
722
+ else:
723
+ self.dtype = dtype
724
725
def add_input_info(self, model_inputs: Dict, force_batch_one: bool = False):
726
self.input_info = [
0 commit comments