Merge pull request #209 from patrykwegrzyn/main

feat: set numCtx = 32768 for Ollama models
This commit is contained in:
Cole Medin
2024-11-07 18:12:27 -06:00
committed by GitHub

View File

@@ -58,7 +58,10 @@ export function getGroqModel(apiKey: string, model: string) {
}
export function getOllamaModel(baseURL: string, model: string) {
let Ollama = ollama(model);
let Ollama = ollama(model, {
numCtx: 32768,
});
Ollama.config.baseURL = `${baseURL}/api`;
return Ollama;
}