provide missed params to data-aware cli

eaidova · eaidova · commit 2b40584e0192 · 2025-01-17T12:15:25.000+04:00
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)
 

Original file line number	Diff line number	Diff line change
`@@ -452,6 +452,8 @@ def run(self):`
`452`	`452`	`quantization_config=quantization_config,`
`453`	`453`	`stateful=not self.args.disable_stateful,`
`454`	`454`	`trust_remote_code=self.args.trust_remote_code,`
	`455`	`+ variant=self.args.variant,`
	`456`	`+ cache_dir=self.args.cache_dir`
`455`	`457`	`)`
`456`	`458`	`model.save_pretrained(self.args.output)`
`457`	`459`