diff --git a/optimum/commands/export/openvino.py b/optimum/commands/export/openvino.py index 37bcb77c4f..0eaddddef4 100644 --- a/optimum/commands/export/openvino.py +++ b/optimum/commands/export/openvino.py @@ -13,7 +13,6 @@ # limitations under the License. """Defines the command line for the export with OpenVINO.""" -import subprocess import sys from pathlib import Path from typing import TYPE_CHECKING, Optional @@ -66,7 +65,6 @@ def __init__( def parse_args(parser: "ArgumentParser"): return parse_args_openvino(parser) - def run(self): from ...exporters.openvino.__main__ import main_export @@ -75,7 +73,7 @@ def run(self): output=self.args.output, task=self.args.task, # fp16=self.args.fp16, - #no_post_process=self.args.no_post_process, + # no_post_process=self.args.no_post_process, # framework=self.args.framework, cache_dir=self.args.cache_dir, # trust_remote_code=self.args.trust_remote_code, diff --git a/optimum/commands/register/register_openvino.py b/optimum/commands/register/register_openvino.py index 7ae12fa75b..a1a74abaca 100644 --- a/optimum/commands/register/register_openvino.py +++ b/optimum/commands/register/register_openvino.py @@ -12,8 +12,8 @@ # See the License for the specific language governing permissions and # limitations under the License. -# from ..openvino.base import OVCommand +from ..export import ExportCommand from ..export.openvino import OVExportCommand -REGISTER_COMMANDS = [OVExportCommand] +REGISTER_COMMANDS = [(OVExportCommand, ExportCommand)]