provide missed params to data-aware cli

huggingface · Jan 17, 2025 · 23e298c · 23e298c
1 parent 0c1189a
commit 23e298c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/optimum/commands/export/openvino.py b/optimum/commands/export/openvino.py
@@ -452,6 +452,8 @@ def run(self):
                 quantization_config=quantization_config,
                 stateful=not self.args.disable_stateful,
                 trust_remote_code=self.args.trust_remote_code,
+                variant=self.args.variant,
+                cache_dir=self.args.cache_dir,
             )
             model.save_pretrained(self.args.output)