Merge pull request #30 from patelvivekdev/main

Added IBM Granite models
sgomez · Oct 27, 2024 · 725b05f · 725b05f
2 parents e0939fe + 5b74f2b
commit 725b05f
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'