Skip to content

Commit 7b5a0bc

Browse files
committed
Update docstring
1 parent d888c24 commit 7b5a0bc

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

extension/android/src/main/java/org/pytorch/executorch/LlamaModule.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,7 @@ public native int generate(
103103
* @param channels Input image number of channels
104104
* @param startPos The starting position in KV cache of the input in the LLM.
105105
* @return The updated starting position in KV cache of the input in the LLM.
106+
* @throws RuntimeException if the prefill failed
106107
*/
107108
public long prefillImages(int[] image, int width, int height, int channels, long startPos) {
108109
long[] nativeResult = prefillImagesNative(image, width, height, channels, startPos);
@@ -125,6 +126,7 @@ private native long[] prefillImagesNative(
125126
* @param bos The number of BOS (begin of sequence) token.
126127
* @param eos The number of EOS (end of sequence) token.
127128
* @return The updated starting position in KV cache of the input in the LLM.
129+
* @throws RuntimeException if the prefill failed
128130
*/
129131
public long prefillPrompt(String prompt, long startPos, int bos, int eos) {
130132
long[] nativeResult = prefillPromptNative(prompt, startPos, bos, eos);

0 commit comments

Comments
 (0)