Improved support for gpt-3.5-turbo-16k

This commit is contained in:
Paul Gauthier 2023-06-13 12:19:56 -07:00
parent e468763c7d
commit 25a28e6f9c
5 changed files with 19 additions and 20 deletions

View file

@ -27,15 +27,14 @@ class Model_GPT35_16k:
max_context_tokens = 16 * 1024
GPT35 = Model_GPT35_16k()
GPT35_16k = Model_GPT35_16k()
GPT35_models = [GPT35, GPT35_16k]
GPT4_models = [GPT4, GPT4_32k]
def get_model(name):
models = [
GPT4_32k,
GPT4,
GPT35,
]
models = GPT35_models + GPT4_models
for model in models:
if model.name == name: