optional

optimum/commands/__init__.py

@@ -12,8 +12,20 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from transformers.utils import is_tf_available
+
+from ..utils import is_onnx_available, is_onnxruntime_available
 from .base import BaseOptimumCLICommand, CommandInfo, RootOptimumCLICommand
 from .env import EnvironmentCommand
-from .export import ExportCommand, ONNXExportCommand, TFLiteExportCommand
-from .onnxruntime import ONNXRuntimeCommand, ONNXRuntimmeOptimizeCommand, ONNXRuntimmeQuantizeCommand
+from .export import ExportCommand
 from .optimum_cli import register_optimum_cli_subcommand
+
+
+if is_onnx_available():
+    from .export import ONNXExportCommand
+
+if is_tf_available():
+    from .export import TFLiteExportCommand
+
+if is_onnxruntime_available():
+    from .onnxruntime import ONNXRuntimeCommand, ONNXRuntimmeOptimizeCommand, ONNXRuntimmeQuantizeCommand

optimum/commands/export/__init__.py

@@ -12,7 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from transformers.utils import is_tf_available
 
+from ...utils import is_onnx_available
 from .base import ExportCommand
-from .onnx import ONNXExportCommand
-from .tflite import TFLiteExportCommand
+
+
+if is_onnx_available():
+    from .onnx import ONNXExportCommand
+if is_tf_available():
+    from .tflite import TFLiteExportCommand

optimum/commands/export/base.py

@@ -14,25 +14,36 @@
 # limitations under the License.
 """optimum.exporters command-line interface base classes."""
 
+from transformers.utils import is_tf_available
+
+from ...utils import is_onnx_available
 from .. import BaseOptimumCLICommand, CommandInfo
-from .onnx import ONNXExportCommand
-from .tflite import TFLiteExportCommand
 
 
 class ExportCommand(BaseOptimumCLICommand):
     COMMAND = CommandInfo(
         name="export",
         help="Export PyTorch and TensorFlow models to several format.",
     )
-    SUBCOMMANDS = (
-        CommandInfo(
-            name="onnx",
-            help="Export PyTorch and TensorFlow to ONNX.",
-            subcommand_class=ONNXExportCommand,
-        ),
-        CommandInfo(
-            name="tflite",
-            help="Export TensorFlow to TensorFlow Lite.",
-            subcommand_class=TFLiteExportCommand,
-        ),
-    )
+    SUBCOMMANDS = ()
+    if is_onnx_available():
+        from .onnx import ONNXExportCommand
+
+        SUBCOMMANDS += (
+            CommandInfo(
+                name="onnx",
+                help="Export PyTorch and TensorFlow to ONNX.",
+                subcommand_class=ONNXExportCommand,
+            ),
+        )
+
+    if is_tf_available():
+        from .tflite import TFLiteExportCommand
+
+        SUBCOMMANDS += (
+            CommandInfo(
+                name="tflite",
+                help="Export TensorFlow to TensorFlow Lite.",
+                subcommand_class=TFLiteExportCommand,
+            ),
+        )

optimum/commands/optimum_cli.py

@@ -17,21 +17,24 @@
 from pathlib import Path
 from typing import Dict, List, Optional, Tuple, Type, Union
 
-from ..utils import logging
+from ..utils import is_onnxruntime_available, logging
 from .base import BaseOptimumCLICommand, CommandInfo, RootOptimumCLICommand
 from .env import EnvironmentCommand
 from .export import ExportCommand
-from .onnxruntime import ONNXRuntimeCommand
 
 
 logger = logging.get_logger()
 
 OPTIMUM_CLI_SUBCOMMANDS = [
     ExportCommand,
     EnvironmentCommand,
-    ONNXRuntimeCommand,
 ]
 
+if is_onnxruntime_available():
+    from .onnxruntime import ONNXRuntimeCommand
+
+    OPTIMUM_CLI_SUBCOMMANDS.append(ONNXRuntimeCommand)
+
 
 def resolve_command_to_command_instance(
     root: RootOptimumCLICommand, commands: List[Type[BaseOptimumCLICommand]]

optimum/utils/__init__.py

@@ -29,6 +29,7 @@
     check_if_transformers_greater,
     is_accelerate_available,
     is_diffusers_available,
+    is_onnx_available,
     is_onnxruntime_available,
     is_pydantic_available,
     is_torch_onnx_support_available,