update

kirklandsign · kirklandsign · commit dd656b082531 · 2025-07-30T23:01:50.000-07:00
diff --git a/extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModule.java b/extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModule.java
@@ -227,7 +227,7 @@ public long prefillPrompt(String prompt, long startPos, int bos, int eos) {
   public int generateFromPos(
       String prompt, int seqLen, long startPos, LlmCallback callback, boolean echo) {
     long[] nativeResult = generateFromPosNative(prompt, seqLen, startPos, callback, echo);
-    return nativeResult[0];
+    return (int) nativeResult[0];
   }
 
   /**

Original file line number	Diff line number	Diff line change
`@@ -227,7 +227,7 @@ public long prefillPrompt(String prompt, long startPos, int bos, int eos) {`
`227`	`227`	`public int generateFromPos(`
`228`	`228`	`String prompt, int seqLen, long startPos, LlmCallback callback, boolean echo) {`
`229`	`229`	`long[] nativeResult = generateFromPosNative(prompt, seqLen, startPos, callback, echo);`
`230`		`- return nativeResult[0];`
	`230`	`+ return (int) nativeResult[0];`
`231`	`231`	`}`
`232`	`232`
`233`	`233`	`/**`