Merge pull request #4424 from BerriAI/fix_add_ollama_codegemma

fix add ollama codegemma to model cost map
This commit is contained in:
Ishaan Jaff 2024-06-26 12:58:27 -07:00 committed by GitHub
commit a631d3a9dc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 18 additions and 0 deletions

View file

@ -3369,6 +3369,15 @@
"supports_function_calling": true,
"supports_parallel_function_calling": true
},
"ollama/codegemma": {
"max_tokens": 8192,
"max_input_tokens": 8192,
"max_output_tokens": 8192,
"input_cost_per_token": 0.0,
"output_cost_per_token": 0.0,
"litellm_provider": "ollama",
"mode": "completion"
},
"ollama/llama2": {
"max_tokens": 4096,
"max_input_tokens": 4096,

View file

@ -3369,6 +3369,15 @@
"supports_function_calling": true,
"supports_parallel_function_calling": true
},
"ollama/codegemma": {
"max_tokens": 8192,
"max_input_tokens": 8192,
"max_output_tokens": 8192,
"input_cost_per_token": 0.0,
"output_cost_per_token": 0.0,
"litellm_provider": "ollama",
"mode": "completion"
},
"ollama/llama2": {
"max_tokens": 4096,
"max_input_tokens": 4096,