diff --git a/opencompass/models/openai_api.py b/opencompass/models/openai_api.py index 692edcf1..9a67c87d 100644 --- a/opencompass/models/openai_api.py +++ b/opencompass/models/openai_api.py @@ -25,7 +25,7 @@ OPENAI_API_BASE = os.path.join( OPENAISDK_API_BASE = os.environ.get('OPENAI_BASE_URL', 'https://api.openai.com/v1/') -O1_MODEL_LIST = ['o1', 'o3'] +O1_MODEL_LIST = ['o1', 'o3', 'o4'] @MODELS.register_module()