From dd250e249eae18b7bdc0639817b1e81a9f96a8c6 Mon Sep 17 00:00:00 2001
From: Hansong Zhang <hsz@meta.com>
Date: Fri, 19 Apr 2024 11:45:16 -0700
Subject: [PATCH] Adding .model tokenizer to selection

Summary: We should allow both .bin and .model for tokenizer

Differential Revision: D56365079
---
 .../com/example/executorchllamademo/MainActivity.java     | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java b/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java
index f2529eb6b86..997d9f87f92 100644
--- a/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java
+++ b/examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java
@@ -119,6 +119,10 @@ private String memoryInfo() {
   private void modelDialog() {
     String[] pteFiles = listLocalFile("/data/local/tmp/llama/", ".pte");
     String[] binFiles = listLocalFile("/data/local/tmp/llama/", ".bin");
+    String[] modelFiles = listLocalFile("/data/local/tmp/llama/", ".model");
+    String[] tokenizerFiles = new String[binFiles.length + modelFiles.length];
+    System.arraycopy(binFiles, 0, tokenizerFiles, 0, binFiles.length);
+    System.arraycopy(modelFiles, 0, tokenizerFiles, binFiles.length, modelFiles.length);
     AlertDialog.Builder modelPathBuilder = new AlertDialog.Builder(this);
     modelPathBuilder.setTitle("Select model path");
     AlertDialog.Builder tokenizerPathBuilder = new AlertDialog.Builder(this);
@@ -134,10 +138,10 @@ private void modelDialog() {
         });
 
     tokenizerPathBuilder.setSingleChoiceItems(
-        binFiles,
+        tokenizerFiles,
         -1,
         (dialog, item) -> {
-          mTokenizerFilePath = binFiles[item];
+          mTokenizerFilePath = tokenizerFiles[item];
           Runnable runnable =
               new Runnable() {
                 @Override