From 6e2e020428a7d33e87dfe1254b2ff83f8eef1494 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 11 Oct 2023 10:18:58 -0700 Subject: [PATCH] (docs) ollama add mistral, code llama and llama2 --- docs/my-website/docs/providers/ollama.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/docs/my-website/docs/providers/ollama.md b/docs/my-website/docs/providers/ollama.md index 24ddbbbf2a..6bccfedd43 100644 --- a/docs/my-website/docs/providers/ollama.md +++ b/docs/my-website/docs/providers/ollama.md @@ -59,15 +59,18 @@ import asyncio asyncio.run(async_ollama()) ``` -### Ollama Models +## Ollama Models Ollama supported models: https://github.com/jmorganca/ollama | Model Name | Function Call | |----------------------|----------------------------------------------------------------------------------- +| Mistral | `completion(model='ollama/mistral', messages, api_base="http://localhost:11434", stream=True)` | | Llama2 7B | `completion(model='ollama/llama2', messages, api_base="http://localhost:11434", stream=True)` | | Llama2 13B | `completion(model='ollama/llama2:13b', messages, api_base="http://localhost:11434", stream=True)` | | Llama2 70B | `completion(model='ollama/llama2:70b', messages, api_base="http://localhost:11434", stream=True)` | | Llama2 Uncensored | `completion(model='ollama/llama2-uncensored', messages, api_base="http://localhost:11434", stream=True)` | +| Code Llama | `completion(model='ollama/codellama', messages, api_base="http://localhost:11434", stream=True)` | +| Llama2 Uncensored | `completion(model='ollama/llama2-uncensored', messages, api_base="http://localhost:11434", stream=True)` | | Orca Mini | `completion(model='ollama/orca-mini', messages, api_base="http://localhost:11434", stream=True)` | | Vicuna | `completion(model='ollama/vicuna', messages, api_base="http://localhost:11434", stream=True)` | | Nous-Hermes | `completion(model='ollama/nous-hermes', messages, api_base="http://localhost:11434", stream=True)` |