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)