Skip to content

Commit 059c729

Browse files
authored
Fix android build (#17126)
1 parent 593775b commit 059c729

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

  • extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm

extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModule.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,7 @@ public LlmModule(
6969
}
7070

7171
mHybridData =
72-
initHybrid(
73-
modelType, modulePath, tokenizerPath, temperature, dataFiles, numBos, numEos);
72+
initHybrid(modelType, modulePath, tokenizerPath, temperature, dataFiles, numBos, numEos);
7473
}
7574

7675
/**
@@ -235,7 +234,8 @@ public int generate(String prompt, LlmCallback llmCallback, boolean echo) {
235234
* @param echo indicate whether to echo the input prompt or not (text completion vs chat)
236235
*/
237236
public int generate(String prompt, int seqLen, LlmCallback llmCallback, boolean echo) {
238-
return generate(prompt, seqLen, llmCallback, echo, DEFAULT_TEMPERATURE);
237+
return generate(
238+
prompt, seqLen, llmCallback, echo, DEFAULT_TEMPERATURE, DEFAULT_BOS, DEFAULT_EOS);
239239
}
240240

241241
/**

0 commit comments

Comments
 (0)