Added IBM Granite models

sgomez · Oct 21, 2024 · 5b74f2b · 5b74f2b
1 parent e0939fe
commit 5b74f2b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/packages/ollama/src/ollama-chat-settings.ts b/packages/ollama/src/ollama-chat-settings.ts
@@ -27,6 +27,12 @@ export type OllamaChatModelId =
   | 'gemma2:2b'
   | 'gemma2:9b'
   | 'gemma2:27b'
+  | 'granite3-dense'
+  | 'granite3-dense:2b'
+  | 'granite3-dense:8b'
+  | 'granite3-moe'
+  | 'granite3-moe:1b'
+  | 'granite3-moe:3b'
   | 'llama2'
   | 'llama2:7b'
   | 'llama2:13b'