Skip to content

Commit 3a50a0b

Browse files
committed
Revert "Arm backend: Deprecate internal models using aot_arm_compiler (pytorch#15302)"
This reverts commit cea66e3.
1 parent efae4a8 commit 3a50a0b

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

examples/arm/aot_arm_compiler.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -91,11 +91,6 @@ 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-
9994
if model_name not in MODELS:
10095
return None
10196

@@ -447,7 +442,7 @@ def get_args():
447442
"-m",
448443
"--model_name",
449444
required=True,
450-
help=f"Model file .py/.pth/.pt or a model from examples/models. Valid names: {set(MODEL_NAME_TO_MODEL.keys())}",
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()))}",
451446
)
452447
parser.add_argument(
453448
"--model_input",

0 commit comments

Comments
 (0)