diff --git a/benchmark/over_time.py b/benchmark/over_time.py index 3955c563a..3d0bb9ae4 100644 --- a/benchmark/over_time.py +++ b/benchmark/over_time.py @@ -26,6 +26,8 @@ def get_legend_label(model): return "DeepSeek" if "mistral" in model: return "Mistral" + if "o1-preview" in model: + return "o1-preview" return model @@ -50,6 +52,9 @@ def get_model_color(model): if "sonnet" in model.lower(): return "orange" + if "o1-preview" in model.lower(): + return "magenta" + if "-4o" in model: return "purple"