From 39b641702c58e2fb9563d5c47048eca168fd0144 Mon Sep 17 00:00:00 2001
From: Anthony Shoumikhin <anthony@shoumikh.in>
Date: Fri, 22 Aug 2025 16:36:57 -0700
Subject: [PATCH] Set an empty EValue input for models that expect None arg.

---
 extension/runner_util/inputs.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/extension/runner_util/inputs.cpp b/extension/runner_util/inputs.cpp
index 842ba25532f..df3727b77d9 100644
--- a/extension/runner_util/inputs.cpp
+++ b/extension/runner_util/inputs.cpp
@@ -55,6 +55,14 @@ Result<BufferCleanup> prepare_input_tensors(
       BufferCleanup cleanup({inputs, num_allocated});
       return tag.error();
     }
+    if (tag.get() == Tag::None) {
+      Error err = method.set_input(runtime::EValue(), i);
+      if (err != Error::Ok) {
+        BufferCleanup cleanup({inputs, num_allocated});
+        return err;
+      }
+      continue;
+    }
     if (tag.get() != Tag::Tensor) {
       ET_LOG(Debug, "Skipping non-tensor input %zu", i);
       continue;