aljaz/chatgpt-telegram-bot
1
1
Fork 0
mirror of https://github.com/aljazceru/chatgpt-telegram-bot.git synced 2025-12-23 23:55:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixed max tokens for gpt-4-128k

Browse Source
This commit is contained in:
AlexHTW
2023-11-07 08:14:58 +01:00
parent f38acff5d4
commit 1f470f4e4a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/openai_helper.py
View File

@@ -42,7 +42,7 @@ def default_max_tokens(model: str) -> int:
elif model in GPT_4_32K_MODELS:
return base * 8
elif model in GPT_4_128K_MODELS:
return base * 100
return base * 31
def are_functions_available(model: str) -> bool:
@@ -418,7 +418,7 @@ class OpenAIHelper:
if self.config['model'] in GPT_4_32K_MODELS:
return base * 8
if self.config['model'] in GPT_4_128K_MODELS:
return base * 100
return base * 31
raise NotImplementedError(
f"Max tokens for model {self.config['model']} is not implemented yet."
)