Merge pull request #1065 from dceoy/feature/add-groq-models

Add Llama 3.1 and Mixtral 8x7B for Groq
Codium-ai · Jul 28, 2024 · 49f608c · 49f608c
2 parents 11f85ca + 495e2cc
commit 49f608c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
@@ -47,5 +47,9 @@
  'bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0': 100000,
  'groq/llama3-8b-8192': 8192,
  'groq/llama3-70b-8192': 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,
  'ollama/llama3': 4096,
 }