zhangshuai/mlx-swift-examples
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update qwen2 chat template (#18)

Browse Source
This commit is contained in:
Madroid Ma
2024-03-07 23:51:54 +08:00
committed by GitHub
parent c4a1e1d108
commit e876e18605
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Libraries/LLM/Models.swift
View File

@@ -89,7 +89,7 @@ extension ModelConfiguration {
id: "mlx-community/Qwen1.5-0.5B-Chat-4bit", id: "mlx-community/Qwen1.5-0.5B-Chat-4bit",
overrideTokenizer: "PreTrainedTokenizer" overrideTokenizer: "PreTrainedTokenizer"
) { prompt in ) { 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 { private enum BootstrapState {