41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
# Copyright (C) 2018-2025 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import sys
|
|
|
|
try:
|
|
import openvino_telemetry as tm
|
|
from openvino_telemetry.backend import backend_ga4
|
|
except ImportError:
|
|
import openvino.tools.ovc.telemetry_stub as tm
|
|
from openvino.tools.ovc.convert_impl import _convert
|
|
from openvino.tools.ovc.cli_parser import get_model_name_from_args
|
|
from openvino.tools.ovc.utils import import_openvino_tokenizers
|
|
|
|
# TODO 131000: temporal workaround to patch OpenVINO Core and frontends with tokenizers extensions
|
|
# make OVC tool to convert models requiring openvino-tokenizers extensions
|
|
import_openvino_tokenizers()
|
|
|
|
# pylint: disable=no-name-in-module,import-error
|
|
from openvino import save_model
|
|
|
|
|
|
def main():
|
|
from openvino.tools.ovc.cli_parser import get_all_cli_parser
|
|
ngraph_function, argv = _convert(get_all_cli_parser(), {}, False)
|
|
if ngraph_function is None:
|
|
return 1
|
|
|
|
model_path = get_model_name_from_args(argv)
|
|
|
|
compress_to_fp16 = 'compress_to_fp16' in argv and argv.compress_to_fp16
|
|
save_model(ngraph_function, model_path.encode('utf-8'), compress_to_fp16)
|
|
|
|
print('[ SUCCESS ] XML file: {}'.format(model_path))
|
|
print('[ SUCCESS ] BIN file: {}'.format(model_path.replace('.xml', '.bin')))
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|