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

fixed max tokens for gpt-3.5-turbo-1106

Browse Source
This commit is contained in:
AlexHTW
2023-11-07 22:28:23 +01:00
parent cd287c8fe0
commit 1e835d7747
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/openai_helper.py
View File

@@ -34,12 +34,12 @@ def default_max_tokens(model: str) -> int:
""" """
base = 1200 base = 1200
if model in GPT_3_MODELS: if model in GPT_3_MODELS:
if model == "gpt-3.5-turbo-0301":
return 4096
return base return base
elif model in GPT_4_MODELS: elif model in GPT_4_MODELS:
return base * 2 return base * 2
elif model in GPT_3_16K_MODELS: elif model in GPT_3_16K_MODELS:
if model == "gpt-3.5-turbo-0301":
return 4096
return base * 4 return base * 4
elif model in GPT_4_32K_MODELS: elif model in GPT_4_32K_MODELS:
return base * 8 return base * 8