Skip to content

Commit

Permalink
add helper for export models via optimum cli (#2471) (#2474)
Browse files Browse the repository at this point in the history
  • Loading branch information
eaidova authored Oct 25, 2024
1 parent ae963a7 commit 5147df2
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions utils/cmd_helper.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,10 @@ def clone_repo(repo_url: str, revision: str = None, add_to_sys_path: bool = True
return repo_path


def optimum_cli(model_id, output_dir, show_command=True, additionl_args: Dict[str, str] = None):
def optimum_cli(model_id, output_dir, show_command=True, additional_args: Dict[str, str] = None):
export_command = f"optimum-cli export openvino --model {model_id} {output_dir}"
if additionl_args is not None:
for arg, value in additionl_args.items():
if additional_args is not None:
for arg, value in additional_args.items():
export_command += f" --{arg}"
if value:
export_command += f" {value}"
Expand Down

0 comments on commit 5147df2

Please sign in to comment.