Skip to content

Commit 920f612

Browse files
committed
Fix Android demo app for SmolLM3
1 parent 890a3ab commit 920f612

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/ModelType.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,5 @@ public enum ModelType {
1515
LLAVA_1_5,
1616
LLAMA_GUARD_3,
1717
QWEN_3,
18+
SMOLLM_3,
1819
}

examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/PromptFormat.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ public static String getSystemPromptTemplate(ModelType modelType) {
2727
case LLAVA_1_5:
2828
return "USER: ";
2929
case QWEN_3:
30+
case SMOLLM_3:
3031
return "<|im_start|>system\n" + "You are a helpful assistant.\n" + "<|im_end|>\n";
3132
default:
3233
return SYSTEM_PLACEHOLDER;
@@ -45,6 +46,7 @@ public static String getUserPromptTemplate(ModelType modelType, boolean thinking
4546
+ "<|start_header_id|>assistant<|end_header_id|>";
4647

4748
case QWEN_3:
49+
case SMOLLM_3:
4850
return "<|im_start|>user\n"
4951
+ USER_PLACEHOLDER
5052
+ "\n<|im_end|>\n"
@@ -68,6 +70,7 @@ public static String getConversationFormat(ModelType modelType) {
6870
case LLAVA_1_5:
6971
return USER_PLACEHOLDER + " ASSISTANT:";
7072
case QWEN_3:
73+
case SMOLLM_3:
7174
return getUserPromptTemplate(modelType, false) + "<|im_end|>\n";
7275
default:
7376
return USER_PLACEHOLDER;
@@ -85,6 +88,8 @@ public static String getStopToken(ModelType modelType) {
8588
return "</s>";
8689
case QWEN_3:
8790
return "<|endoftext|>";
91+
case SMOLLM_3:
92+
return "<|end_of_text|>";
8893
default:
8994
return "";
9095
}
@@ -93,6 +98,7 @@ public static String getStopToken(ModelType modelType) {
9398
public static String getThinkingModeToken(ModelType modelType, boolean thinkingMode) {
9499
switch (modelType) {
95100
case QWEN_3:
101+
case SMOLLM_3:
96102
return thinkingMode ? "" : "<think>\n\n</think>\n\n\n";
97103
default:
98104
return "";
@@ -102,6 +108,7 @@ public static String getThinkingModeToken(ModelType modelType, boolean thinkingM
102108
public static String replaceSpecialToken(ModelType modelType, String token) {
103109
switch (modelType) {
104110
case QWEN_3:
111+
case SMOLLM_3:
105112
switch (token) {
106113
case "<|im_end|>":
107114
return "";

0 commit comments

Comments
 (0)