update qwen2 chat template (#18)

This commit is contained in:
Madroid Ma authored and GitHub committed 2024-03-07 07:51:54 -08:00
1 parent c4a1e1d108
commit e876e18605
1 file changed
+1 -1
+1 -1
View File
@@ -89,7 +89,7 @@ extension ModelConfiguration {
id: "mlx-community/Qwen1.5-0.5B-Chat-4bit",
overrideTokenizer: "PreTrainedTokenizer"
) { prompt in
"<|im_start|>user \(prompt)<|im_end|><|im_start|>assistant"
"<|im_start|>system\nYou are a helpful assistant<|im_end|>\n<|im_start|>user\n\(prompt)<|im_end|>\n<|im_start|>assistant"
}
private enum BootstrapState {