nemo_automodel.cli.query_capabilities

Thin CLI wrapper around :func:nemo_automodel.query_capabilities.

Usage::

python -m nemo_automodel.cli.query_capabilities —model-id meta-llama/Llama-3.1-8B python -m nemo_automodel.cli.query_capabilities —model-id <id> —json python -m nemo_automodel.cli.query_capabilities —model-id <id> —trust-remote-code

Module Contents

Functions

Name	Description
`_parse_args`	-
`_print_table`	-
`main`	CLI entry point. Returns a process exit code (0 on success).

API

nemo_automodel.cli.query_capabilities._parse_args(
    argv: list[str] | None = None
) -> argparse.Namespace

nemo_automodel.cli.query_capabilities._print_table(
    model_id: str,
    caps: nemo_automodel._transformers.model_capabilities.ModelCapabilities
) -> None

nemo_automodel.cli.query_capabilities.main(
    argv: list[str] | None = None
) -> int

CLI entry point. Returns a process exit code (0 on success).