From 76eb9a1115f4e9e072b1802f7a2ff1b779ccbde8 Mon Sep 17 00:00:00 2001
From: Hansong Zhang <hsz@meta.com>
Date: Tue, 14 Oct 2025 15:35:28 -0700
Subject: [PATCH] Fix JNI signature

---
 extension/android/jni/jni_layer_llama.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/extension/android/jni/jni_layer_llama.cpp b/extension/android/jni/jni_layer_llama.cpp
index fcc69509352..d12783d4cf0 100644
--- a/extension/android/jni/jni_layer_llama.cpp
+++ b/extension/android/jni/jni_layer_llama.cpp
@@ -140,7 +140,8 @@ class ExecuTorchLlmJni : public facebook::jni::HybridClass<ExecuTorchLlmJni> {
       facebook::jni::alias_ref<jstring> model_path,
       facebook::jni::alias_ref<jstring> tokenizer_path,
       jfloat temperature,
-      facebook::jni::alias_ref<jobject> data_files) {
+      facebook::jni::alias_ref<facebook::jni::JList<jstring>::javaobject>
+          data_files) {
     return makeCxxInstance(
         model_type_category,
         model_path,