Pull request pytorch#118: Support Toy models in aot_neutron_compile via --model_name

pietraferreira · pietraferreira · commit ffa1ec50e40b · 2025-04-09T14:51:36.000+02:00
Merge in AITEC/executorch from pf-toymodels-support to main-nxp

* commit '0390a5c78d2342cba8f2507587900166f07c5794':
  Support Toy models in aot_neutron_compile via --model_name
diff --git a/examples/nxp/aot_neutron_compile.py b/examples/nxp/aot_neutron_compile.py
@@ -77,7 +77,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
`@@ -77,7 +77,7 @@ def get_model_and_inputs_from_name(model_name: str):`
`77`	`77`	`logging.warning(`
`78`	`78`	`"Using a model from examples/models not all of these are currently supported"`
`79`	`79`	`)`
`80`		`- model, example_inputs, _ = EagerModelFactory.create_model(*MODEL_NAME_TO_MODEL[model_name])`
	`80`	`+ model, example_inputs, _, _ = EagerModelFactory.create_model(*MODEL_NAME_TO_MODEL[model_name])`
`81`	`81`	`else:`
`82`	`82`	`raise RuntimeError(`
`83`	`83`	`f"Model '{model_name}' is not a valid name. Use --help for a list of available models."`