OpenBMB/ChatDev
1
1
Fork 0
mirror of https://github.com/OpenBMB/ChatDev.git synced 2024-12-24 20:32:07 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

Remove reference to Gemini vision

Browse Source
This commit is contained in:
Eric 2024-10-04 00:08:09 -04:00
parent f42f11b353
commit a83bb998f7
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
camel/model_backend.py
View File

@ -175,8 +175,7 @@ class GeminiModel(ModelBackend):
num_prompt_tokens = model.count_tokens(string).total_tokens
num_max_token_map = {
"gemini-1.5-pro": 1000000,
"gemini-1.5-flash": 1000000,
"gemini-pro-vision": 12288,
"gemini-1.5-flash": 1000000
}
num_max_token = num_max_token_map[self.model_type.value]
num_max_completion_tokens = num_max_token - num_prompt_tokens