From 71c558d3064636f710e271a977cdb30657a9c23f Mon Sep 17 00:00:00 2001 From: dceoy Date: Sat, 28 Dec 2024 01:51:33 +0900 Subject: [PATCH] Update Groq models and tokens --- pr_agent/algo/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index cfa1bbe3..b092077a 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -66,10 +66,10 @@ MAX_TOKENS = { 'claude-3-5-sonnet': 100000, 'groq/llama3-8b-8192': 8192, 'groq/llama3-70b-8192': 8192, + 'groq/llama-3.1-8b-instant': 8192, + 'groq/llama-3.3-70b-versatile': 8192, 'groq/mixtral-8x7b-32768': 32768, - 'groq/llama-3.1-8b-instant': 131072, - 'groq/llama-3.1-70b-versatile': 131072, - 'groq/llama-3.1-405b-reasoning': 131072, + 'groq/gemma2-9b-it': 8192, 'ollama/llama3': 4096, 'watsonx/meta-llama/llama-3-8b-instruct': 4096, "watsonx/meta-llama/llama-3-70b-instruct": 4096,