fix

huggingface · Sep 27, 2023 · 66e7006 · 66e7006
1 parent e7ae52a
commit 66e7006
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git 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
@@ -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)]