From f50832e19b72218342adcea4b15a1e219fd7fcb7 Mon Sep 17 00:00:00 2001 From: Tal Date: Mon, 29 Jul 2024 08:32:34 +0300 Subject: [PATCH] Update __init__.py --- pr_agent/algo/__init__.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 27b24c56..baea8a3a 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -52,4 +52,10 @@ MAX_TOKENS = { 'groq/llama-3.1-70b-versatile': 131072, 'groq/llama-3.1-405b-reasoning': 131072, 'ollama/llama3': 4096, + 'watsonx/meta-llama/llama-3-8b-instruct': 4096, + "watsonx/meta-llama/llama-3-70b-instruct": 4096, + "watsonx/meta-llama/llama-3-405b-instruct": 16384, + "watsonx/ibm/granite-13b-chat-v2": 8191, + "watsonx/ibm/granite-34b-code-instruct": 8191, + "watsonx/mistralai/mistral-large": 32768, }