From c30fa2be636406658066fc8f9c71ac4226fc26dd Mon Sep 17 00:00:00 2001 From: Lucy Qiu Date: Thu, 19 Mar 2026 16:48:49 -0700 Subject: [PATCH] Update [ghstack-poisoned] --- examples/apple/coreml/llama/run_static_llm.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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(