ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-15 12:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Update __init__.py

Browse source
This commit is contained in:
Tal 2024-07-29 08:32:34 +03:00 committed by GitHub
parent 927f124dca
commit f50832e19b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pr_agent/algo/__init__.py
View file

@ -52,4 +52,10 @@ MAX_TOKENS = {
'groq/llama-3.1-70b-versatile': 131072, 'groq/llama-3.1-70b-versatile': 131072,
'groq/llama-3.1-405b-reasoning': 131072, 'groq/llama-3.1-405b-reasoning': 131072,
'ollama/llama3': 4096, '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,
} }