From d3025cfd19792a7137922cc9a3f0b64cbcd4eec7 Mon Sep 17 00:00:00 2001
From: zwcolin <ziruiw2000@gmail.com>
Date: Mon, 19 Aug 2024 15:30:27 -0400
Subject: [PATCH] add support for `chatgpt-4o-latest`
---
src/generate_lib/utils.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/src/generate_lib/utils.py b/src/generate_lib/utils.py
index a127b49..efcd6d0 100644
--- a/src/generate_lib/utils.py
+++ b/src/generate_lib/utils.py
@@ -14,7 +14,8 @@ def generate_response_remote_wrapper(generate_fn,
try:
result = generate_fn(image, query, model_path,
api_key=api_key, client=client, random_baseline=False)
- except:
+ except Exception as e:
+ print(f"Error: {e}")
print(f"Error {curr_retries}, sleeping for {sleep_time} seconds...")
time.sleep(sleep_time)
curr_retries += 1
@@ -39,6 +40,7 @@ def get_client_fn(model_path):
# gpt
elif model_path in ['gpt-4o-2024-05-13',
'gpt-4o-2024-08-06',
+ 'chatgpt-4o-latest',
'gpt-4-turbo-2024-04-09',
'gpt-4o-mini-2024-07-18']:
from .gpt import get_client_model
@@ -84,6 +86,7 @@ def get_generate_fn(model_path):
# gpt
elif model_name in ['gpt-4o-2024-05-13',
'gpt-4o-2024-08-06',
+ 'chatgpt-4o-latest',
'gpt-4-turbo-2024-04-09',
'gpt-4o-mini-2024-07-18']:
from .gpt import generate_response
--
GitLab