Skip to content

Commit 4107ea3

Browse files
authored
Add support for qwen2 model in llm iOS app (#112)
1 parent 518e197 commit 4107ea3

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

llm/apple/Application/ContentView.swift

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,7 @@ struct ContentView: View {
212212
case gemma3
213213
case llama
214214
case llava
215+
case qwen2
215216
case qwen3
216217
case phi4
217218
case smollm3
@@ -223,6 +224,8 @@ struct ContentView: View {
223224
return .gemma3
224225
} else if filename.hasPrefix("llama") {
225226
return .llama
227+
} else if filename.hasPrefix("qwen2") {
228+
return .qwen2
226229
} else if filename.hasPrefix("llava") {
227230
return .llava
228231
} else if filename.hasPrefix("qwen3") {
@@ -712,6 +715,7 @@ struct ContentView: View {
712715
case .llama: return .llamagenerated
713716
case .llava: return .llavagenerated
714717
case .phi4: return .phi4generated
718+
case .qwen2: return .qwengenerated
715719
case .qwen3: return .qwengenerated
716720
case .smollm3: return .smollm3generated
717721
case .voxtral: return .voxtralgenerated
@@ -937,6 +941,8 @@ struct ContentView: View {
937941
formattedPrompt = String(format: Constants.llavaPromptTemplate, text)
938942
case .phi4:
939943
formattedPrompt = String(format: Constants.phi4PromptTemplate, text)
944+
case .qwen2:
945+
formattedPrompt = String(format: Constants.llama3PromptTemplate, text)
940946
case .qwen3:
941947
let basePrompt = String(format: Constants.qwen3PromptTemplate, text)
942948
formattedPrompt = thinkingMode ? basePrompt.replacingOccurrences(of: "<think>\n\n</think>\n\n\n", with: "") : basePrompt

0 commit comments

Comments
 (0)