From 16d7082e59cde6d449d00c266edb41a583a709fa Mon Sep 17 00:00:00 2001 From: Riandy Riandy Date: Thu, 5 Sep 2024 11:08:04 -0700 Subject: [PATCH] Update Android app to support echo flag in generate Summary: Echo flag is added in the runner side via https://github.com/pytorch/executorch/pull/5011 by Chirag. Now, we update the app side to leverage the new echo flag, so that we don't display the user prompt in response. Differential Revision: D62250116 --- .../java/com/example/executorchllamademo/MainActivity.java | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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 f24254efb31..0d8553bdb4c 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 @@ -614,6 +614,7 @@ public void run() { ModelUtils.VISION_MODEL_IMAGE_CHANNELS, prompt, ModelUtils.VISION_MODEL_SEQ_LEN, + false, MainActivity.this); } else { // no image selected, we pass in empty int array @@ -624,10 +625,12 @@ public void run() { ModelUtils.VISION_MODEL_IMAGE_CHANNELS, prompt, ModelUtils.VISION_MODEL_SEQ_LEN, + false, MainActivity.this); } } else { - mModule.generate(prompt, ModelUtils.TEXT_MODEL_SEQ_LEN, MainActivity.this); + mModule.generate( + prompt, ModelUtils.TEXT_MODEL_SEQ_LEN, false, MainActivity.this); } long generateDuration = System.currentTimeMillis() - generateStartTime;