diff --git a/examples/apple/coreml/llama/run_static_llm.py b/examples/apple/coreml/llama/run_static_llm.py index 107448b3453..a2a12d56981 100644 --- a/examples/apple/coreml/llama/run_static_llm.py +++ b/examples/apple/coreml/llama/run_static_llm.py @@ -171,6 +171,12 @@ def main(): default=4, help="Number of verification branches for lookahead decoding", ) + parser.add_argument( + "--method", + type=str, + default="forward", + help="Method name to execute (default: forward)", + ) args = parser.parse_args() @@ -208,7 +214,7 @@ def main(): print(f"Loading model from {args.model}...") runtime = Runtime.get() program = runtime.load_program(args.model) - method = program.load_method("forward") + method = program.load_method(args.method) metadata = method.metadata print(