This commit is contained in:
Paul Gauthier 2023-06-14 17:10:27 -07:00
parent 67ee800c55
commit 8c91e107e0

View file

@ -27,4 +27,15 @@ def get_model(name):
if model.name == name:
return model
tokens = 0
model = Model(name, tokens)
if name.startswith("gpt-4-"):
GPT4_models.append(model)
elif name.startswith("gpt-3.5-"):
GPT35_models.append(model)
else:
raise ValueError(f"Unsupported model: {name}")
return model