nemo_automodel.cli.utils#

Module Contents#

Functions#

_discover_recipe_classes

Scan nemo_automodel/recipes/ for concrete recipe classes.

resolve_recipe_name

Resolve a recipe name to its fully-qualified dotted path.

load_yaml

Load and return a YAML file as a dict.

Data#

_RECIPES_DIR

logger

API#

nemo_automodel.cli.utils._RECIPES_DIR#

None

nemo_automodel.cli.utils.logger#

‘getLogger(…)’

nemo_automodel.cli.utils._discover_recipe_classes() → dict[str, str][source]#

Scan nemo_automodel/recipes/ for concrete recipe classes.

Returns a mapping from bare class name to fully-qualified dotted path, e.g. {"TrainFinetuneRecipeForNextTokenPrediction": "nemo_automodel.recipes.llm.train_ft.TrainFinetuneRecipeForNextTokenPrediction"}.

nemo_automodel.cli.utils.resolve_recipe_name(raw: str) → str[source]#

Resolve a recipe name to its fully-qualified dotted path.

Accepts:

  • Bare class name: "TrainFinetuneRecipeForNextTokenPrediction"

  • Full FQN: "nemo_automodel.recipes.llm.train_ft.TrainFinetuneRecipeForNextTokenPrediction"

Raises ValueError when a bare name cannot be found.

nemo_automodel.cli.utils.load_yaml(file_path)[source]#

Load and return a YAML file as a dict.