Raise error if dataset is not provided for full quantization via opti…

…mum-cli
huggingface · Jan 17, 2025 · e2f3512 · e2f3512
1 parent 2590794
commit e2f3512
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/optimum/commands/export/openvino.py b/optimum/commands/export/openvino.py
@@ -364,6 +364,11 @@ def run(self):
                 quantization_config["trust_remote_code"] = self.args.trust_remote_code
             ov_config = OVConfig(quantization_config=quantization_config)
         else:
+            if self.args.dataset is None:
+                raise ValueError(
+                    "Dataset is required for full quantization. Please provide it with --dataset argument."
+                )
+
             quantization_config = {
                 "weight_format": self.args.quant_mode,
                 "activation_format": self.args.quant_mode,