From 2b40584e01929866f549d43c1bc8816b12d40c05 Mon Sep 17 00:00:00 2001 From: eaidova Date: Fri, 17 Jan 2025 12:15:25 +0400 Subject: [PATCH] provide missed params to data-aware cli --- optimum/commands/export/openvino.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/optimum/commands/export/openvino.py b/optimum/commands/export/openvino.py index 161534fad..554725738 100644 --- 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)