minor: add gpt-4o ctx size annotation

evalplus · May 19, 2024 · fc34193 · fc34193
1 parent 2a3557a
commit fc34193
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/repoqa/compute_score.py b/repoqa/compute_score.py
@@ -328,7 +328,7 @@ def compute_score(
             train_context = "16k"
         else:
             train_context = fetch_hf_context(model_name)
-    elif model_name.startswith("gpt-4-turbo"):
+    elif model_name.startswith("gpt-4-turbo") or model_name.startswith("gpt-4o-"):
         train_context = "128k"
     elif model_name.startswith("gpt-3.5-"):
         train_context = "16k"