diff options
Diffstat (limited to 'src/mlia/cli/main.py')
-rw-r--r-- | src/mlia/cli/main.py | 15 |
1 files changed, 5 insertions, 10 deletions
diff --git a/src/mlia/cli/main.py b/src/mlia/cli/main.py index 2b63124..4a91b08 100644 --- a/src/mlia/cli/main.py +++ b/src/mlia/cli/main.py @@ -11,7 +11,6 @@ from inspect import signature from mlia import __version__ from mlia.backend.errors import BackendUnavailableError -from mlia.backend.registry import registry as backend_registry from mlia.cli.commands import backend_install from mlia.cli.commands import backend_list from mlia.cli.commands import backend_uninstall @@ -35,7 +34,7 @@ from mlia.core.context import ExecutionContext from mlia.core.errors import ConfigurationError from mlia.core.errors import InternalError from mlia.core.logging import setup_logging -from mlia.target.registry import registry as target_registry +from mlia.target.registry import table as target_table logger = logging.getLogger(__name__) @@ -43,14 +42,10 @@ logger = logging.getLogger(__name__) INFO_MESSAGE = f""" ML Inference Advisor {__version__} -Help the design and optimization of neural network models for efficient inference on a target CPU and NPU - -Supported targets: -{target_registry} - -Supported backends: -{backend_registry} +Help the design and optimization of neural network models for efficient inference on a target CPU or NPU. +{target_table().to_plain_text(show_title=True, space=False)} +Use command 'mlia-backend' to install backends. """.strip() @@ -192,7 +187,7 @@ def run_command(args: argparse.Namespace) -> int: ) try: - logger.info(INFO_MESSAGE) + logger.info("ML Inference Advisor %s", __version__) logger.info( "\nThis execution of MLIA uses output directory: %s", ctx.output_dir ) |