From 1e835d7747735452ae0eff749dfa5e7e1901cde5 Mon Sep 17 00:00:00 2001 From: AlexHTW Date: Tue, 7 Nov 2023 22:28:23 +0100 Subject: [PATCH] fixed max tokens for gpt-3.5-turbo-1106 --- bot/openai_helper.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bot/openai_helper.py b/bot/openai_helper.py index 26fad44..9065d9d 100644 --- a/bot/openai_helper.py +++ b/bot/openai_helper.py @@ -34,12 +34,12 @@ def default_max_tokens(model: str) -> int: """ base = 1200 if model in GPT_3_MODELS: - if model == "gpt-3.5-turbo-0301": - return 4096 return base elif model in GPT_4_MODELS: 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 elif model in GPT_4_32K_MODELS: return base * 8