🐛 fix(ollama): fix gemma model id (lobehub#1377)

hanwwh · Feb 24, 2024 · 3da50ff · 3da50ff
1 parent ec76917
commit 3da50ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/config/modelProviders/ollama.ts b/src/config/modelProviders/ollama.ts
@@ -5,14 +5,14 @@ const Ollama: ModelProviderCard = {
     {
       displayName: 'Gemma 7B',
       functionCall: false,
-      id: 'gemma:7b',
+      id: 'gemma',
       tokens: 4000,
       vision: false,
     },
     {
       displayName: 'Gemma 2B',
       functionCall: false,
-      id: 'gemma',
+      id: 'gemma:2b',
       tokens: 4000,
       vision: false,
     },