Skip to content

Commit d0d845a

Browse files
authored
Fix Qwen3 prompt format issues (#152)
1 parent 2052cfe commit d0d845a

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

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

llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/PromptFormat.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ public static String getSystemPromptTemplate(ModelType modelType) {
2323
+ SYSTEM_PLACEHOLDER
2424
+ "<|eot_id|>";
2525
case QWEN_3:
26-
return "<|im_start|>system\n" + "You are a helpful assistant.\n" + "<|im_end|>\n";
26+
return "<|im_start|>system\n" + SYSTEM_PLACEHOLDER + "<|im_end|>\n";
2727
default:
2828
return SYSTEM_PLACEHOLDER;
2929
}
@@ -43,7 +43,7 @@ public static String getUserPromptTemplate(ModelType modelType) {
4343
case QWEN_3:
4444
return "<|im_start|>user\n"
4545
+ USER_PLACEHOLDER
46-
+ "\n<|im_end|>\n"
46+
+ "<|im_end|>\n"
4747
+ "<|im_start|>assistant\n"
4848
+ THINKING_MODE_PLACEHOLDER;
4949
case LLAVA_1_5:
@@ -63,7 +63,7 @@ public static String getStopToken(ModelType modelType) {
6363
case LLAVA_1_5:
6464
return "</s>";
6565
case QWEN_3:
66-
return "<|endoftext|>";
66+
return "<|im_end|>";
6767
case VOXTRAL:
6868
return "</s>";
6969
default:

0 commit comments

Comments
 (0)