Skip to content

Commit 38d944c

Browse files
Style
1 parent 06b2250 commit 38d944c

File tree

2 files changed

+12
-12
lines changed

2 files changed

+12
-12
lines changed

optimum/commands/export/openvino.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -366,13 +366,13 @@ def run(self):
366366
q_config = prepare_q_config(self.args)
367367
q_config["activation_format"] = "f8e4m3"
368368

369-
quantization_config = dict(
370-
weight_quantization_config=wc_config,
371-
activation_quantization_config=q_config,
372-
num_samples=self.args.num_samples,
373-
dataset=self.args.dataset,
374-
trust_remote_code=self.args.trust_remote_code,
375-
)
369+
quantization_config = {
370+
"weight_quantization_config": wc_config,
371+
"activation_quantization_config": q_config,
372+
"num_samples": self.args.num_samples,
373+
"dataset": self.args.dataset,
374+
"trust_remote_code": self.args.trust_remote_code,
375+
}
376376
else:
377377
quantization_config = prepare_q_config(self.args)
378378
ov_config = OVConfig(quantization_config=quantization_config)

optimum/intel/openvino/configuration.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -890,11 +890,11 @@ def __init__(
890890
# Pull dataset-related parameters from child configs. This is not the intended use case, but we process it just
891891
# in case user sets those parameters inside child configs only.
892892
wqc, aqc = self.weight_quantization_config, self.activation_quantization_config
893-
num_samples = num_samples or self.wqc.num_samples or self.aqc.num_samples
894-
dataset = dataset or self.wqc.dataset or self.aqc.dataset
895-
tokenizer = tokenizer or self.wqc.tokenizer or self.aqc.tokenizer
896-
processor = processor or self.wqc.processor or self.aqc.processor
897-
trust_remote_code = trust_remote_code or self.wqc.trust_remote_code or self.aqc.trust_remote_code
893+
num_samples = num_samples or wqc.num_samples or aqc.num_samples
894+
dataset = dataset or wqc.dataset or aqc.dataset
895+
tokenizer = tokenizer or wqc.tokenizer or aqc.tokenizer
896+
processor = processor or wqc.processor or aqc.processor
897+
trust_remote_code = trust_remote_code or wqc.trust_remote_code or aqc.trust_remote_code
898898
super().__init__(
899899
num_samples=num_samples,
900900
dataset=dataset,

0 commit comments

Comments
 (0)