Fix variable number of inputs in mps_example.py

Author: DenisVieriu97

examples/apple/mps/executor_runner/mps_executor_runner.mm

@@ -481,10 +481,11 @@ MemoryManager memory_manager(
         strstr(model_path, "emformer_transcribe")  ||
         strstr(model_path, "emformer_join")        ||
         strstr(model_path, "edsr")                 ||
+        strstr(model_path, "llama2")               ||
         strstr(model_path, "ic3")                  ||
         strstr(model_path, "ic4")) {
         atol = 1e-04;
-      }
+    }
     status = torch::executor::util::VerifyResultWithBundledExpectedOutput(
         *method,
         file_data->data(),

examples/apple/mps/scripts/mps_example.py

@@ -75,8 +75,8 @@ def __init__(self):
             super().__init__()
             self.mps_module = lowered_module
 
-        def forward(self, input_args):
-            return self.mps_module(input_args)
+        def forward(self, *input_args):
+            return self.mps_module(*input_args)
 
     executorch_program = (
         exir.capture(