Skip to content

Commit

Permalink
Fix the is_chat_model function to work with gpt-4o
Browse files Browse the repository at this point in the history
  • Loading branch information
Lorenzo Pacchiardi committed Aug 22, 2024
1 parent 234bcde commit 6a5fc4a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion evals/registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ def is_chat_model(model_name: str) -> bool:
if model_name in CHAT_MODEL_NAMES:
return True

for model_prefix in {"gpt-3.5-turbo-", "gpt-4-"}:
for model_prefix in {"gpt-3.5-turbo-", "gpt-4-", "gpt-4o-"}:
if model_name.startswith(model_prefix):
return True

Expand Down

0 comments on commit 6a5fc4a

Please sign in to comment.