Skip to content

Commit 22f462a

Browse files
fix: Android LlamaDemo app - remove unnecessary stop code check per model (#200)
1 parent 6e135a2 commit 22f462a

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

  • llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/ui/viewmodel

llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/ui/viewmodel/ChatViewModel.kt

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -740,11 +740,7 @@ class ChatViewModel(application: Application) : AndroidViewModel(application), L
740740
var processedResult = result
741741

742742
if (processedResult == PromptFormat.getStopToken(currentSettingsFields.modelType)) {
743-
if (currentSettingsFields.modelType == ModelType.GEMMA_3 ||
744-
currentSettingsFields.modelType == ModelType.LLAVA_1_5
745-
) {
746-
module?.stop()
747-
}
743+
module?.stop()
748744
return
749745
}
750746

0 commit comments

Comments
 (0)