From 291996180378b8740837acd63dfa83a6638f06a4 Mon Sep 17 00:00:00 2001 From: Andrey Churkin Date: Fri, 24 Jan 2025 13:32:32 +0000 Subject: [PATCH] Add nncf.strip call --- tools/ovc/openvino/tools/ovc/convert_impl.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tools/ovc/openvino/tools/ovc/convert_impl.py b/tools/ovc/openvino/tools/ovc/convert_impl.py index 3ad759110b55d8..1529d2acd6c9f9 100644 --- a/tools/ovc/openvino/tools/ovc/convert_impl.py +++ b/tools/ovc/openvino/tools/ovc/convert_impl.py @@ -478,6 +478,12 @@ def _convert(cli_parser: argparse.ArgumentParser, args, python_api_used): get_jax_decoder(args['input_model'], args) else: raise Error("JAX Frontend is not available.") + if model_framework == "tf": + try: + from nncf.tensorflow.strip import strip + args['input_model'] = strip(args['input_model']) + except: + pass argv = pack_params_to_args_namespace(args, cli_parser, python_api_used)