add support for llama vision (gluonfield#165)

Co-authored-by: Mattia <[email protected]>
opoyraz · Nov 7, 2024 · 2aa9d14 · 2aa9d14
1 parent 2207cbb
commit 2aa9d14
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Enchanted/Services/OllamaService.swift b/Enchanted/Services/OllamaService.swift
@@ -40,7 +40,7 @@ class OllamaService: @unchecked Sendable {
             LanguageModel(
                 name: $0.name,
                 provider: .ollama,
-                imageSupport: $0.details.families?.contains("clip") ?? false
+                imageSupport: $0.details.families?.contains(where: { $0 == "clip" || $0 == "mllama" }) ?? false
             )
         }
         return models

diff --git a/Enchanted/UI/macOS/Components/PromptPanelView.swift b/Enchanted/UI/macOS/Components/PromptPanelView.swift
@@ -36,7 +36,7 @@ struct PromptPanelView: View {
 
     var imageSupportMissing: some View {
         HStack {
-            Text("This model does not support images. Supported models are llava and bakllava.")
+            Text("This model does not support images. Supported models are llava, bakllava and llama vision.")
                 .font(.caption2)
             Spacer()
         }