Support Toy models in aot_neutron_compile via --model_name

pietraferreira · pietraferreira · commit 0390a5c78d23 · 2025-03-18T10:59:30.000Z
Toy models can now be selected in `aot_neutron_compile.py` using the `--model_name` flag. This was achieved with a minor change to `get_model_and_inputs_from_name`, ensuring the models are correctly retrieved from `MODEL_NAME_TO_MODEL`.
diff --git a/examples/nxp/aot_neutron_compile.py b/examples/nxp/aot_neutron_compile.py
@@ -75,7 +75,7 @@ def get_model_and_inputs_from_name(model_name: str):
         logging.warning(
             "Using a model from examples/models not all of these are currently supported"
         )
-        model, example_inputs, _ = EagerModelFactory.create_model(*MODEL_NAME_TO_MODEL[model_name])
+        model, example_inputs, _, _ = EagerModelFactory.create_model(*MODEL_NAME_TO_MODEL[model_name])
     else:
         raise RuntimeError(
             f"Model '{model_name}' is not a valid name. Use --help for a list of available models."

Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ def get_model_and_inputs_from_name(model_name: str):`
`75`	`75`	`logging.warning(`
`76`	`76`	`"Using a model from examples/models not all of these are currently supported"`
`77`	`77`	`)`
`78`		`- model, example_inputs, _ = EagerModelFactory.create_model(*MODEL_NAME_TO_MODEL[model_name])`
	`78`	`+ model, example_inputs, _, _ = EagerModelFactory.create_model(*MODEL_NAME_TO_MODEL[model_name])`
`79`	`79`	`else:`
`80`	`80`	`raise RuntimeError(`
`81`	`81`	`f"Model '{model_name}' is not a valid name. Use --help for a list of available models."`