Skip to content

Commit cea66e3

Browse files
Arm backend: Deprecate internal models using aot_arm_compiler (#15302)
Signed-off-by: Sebastian Larsson <[email protected]>
1 parent d6de876 commit cea66e3

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

examples/arm/aot_arm_compiler.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,11 @@ def _load_internal_model(
9191
model_name: str, example_inputs: Any
9292
) -> Optional[Tuple[torch.nn.Module, Any]]:
9393
"""Load a bundled example model from the internal `MODELS` mapping."""
94+
logging.info(
95+
"Loading internal models is deprecated. Use --model_name <FILE>.py/.pt "
96+
"or a model from examples/models."
97+
)
98+
9499
if model_name not in MODELS:
95100
return None
96101

@@ -442,7 +447,7 @@ def get_args():
442447
"-m",
443448
"--model_name",
444449
required=True,
445-
help=f"Model file .py/.pth/.pt, builtin model or a model from examples/models. Valid names: {set(list(MODELS.keys()) + list(MODEL_NAME_TO_MODEL.keys()))}",
450+
help=f"Model file .py/.pth/.pt or a model from examples/models. Valid names: {set(MODEL_NAME_TO_MODEL.keys())}",
446451
)
447452
parser.add_argument(
448453
"--model_input",

0 commit comments

Comments
 (0)