Skip to content

Commit e2f3512

Browse files
Raise error if dataset is not provided for full quantization via optimum-cli
1 parent 2590794 commit e2f3512

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

optimum/commands/export/openvino.py

+5
Original file line numberDiff line numberDiff line change
@@ -364,6 +364,11 @@ def run(self):
364364
quantization_config["trust_remote_code"] = self.args.trust_remote_code
365365
ov_config = OVConfig(quantization_config=quantization_config)
366366
else:
367+
if self.args.dataset is None:
368+
raise ValueError(
369+
"Dataset is required for full quantization. Please provide it with --dataset argument."
370+
)
371+
367372
quantization_config = {
368373
"weight_format": self.args.quant_mode,
369374
"activation_format": self.args.quant_mode,

0 commit comments

Comments
 (0)