From 23e298c06f0acf811478ccc13d435268dda8141b 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..695c22e98 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)