Skip to content
Snippets Groups Projects
Commit d3025cfd authored by zwcolin's avatar zwcolin
Browse files

add support for `chatgpt-4o-latest`

parent 6747cbc9
No related branches found
No related tags found
No related merge requests found
...@@ -14,7 +14,8 @@ def generate_response_remote_wrapper(generate_fn, ...@@ -14,7 +14,8 @@ def generate_response_remote_wrapper(generate_fn,
try: try:
result = generate_fn(image, query, model_path, result = generate_fn(image, query, model_path,
api_key=api_key, client=client, random_baseline=False) 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...") print(f"Error {curr_retries}, sleeping for {sleep_time} seconds...")
time.sleep(sleep_time) time.sleep(sleep_time)
curr_retries += 1 curr_retries += 1
...@@ -39,6 +40,7 @@ def get_client_fn(model_path): ...@@ -39,6 +40,7 @@ def get_client_fn(model_path):
# gpt # gpt
elif model_path in ['gpt-4o-2024-05-13', elif model_path in ['gpt-4o-2024-05-13',
'gpt-4o-2024-08-06', 'gpt-4o-2024-08-06',
'chatgpt-4o-latest',
'gpt-4-turbo-2024-04-09', 'gpt-4-turbo-2024-04-09',
'gpt-4o-mini-2024-07-18']: 'gpt-4o-mini-2024-07-18']:
from .gpt import get_client_model from .gpt import get_client_model
...@@ -84,6 +86,7 @@ def get_generate_fn(model_path): ...@@ -84,6 +86,7 @@ def get_generate_fn(model_path):
# gpt # gpt
elif model_name in ['gpt-4o-2024-05-13', elif model_name in ['gpt-4o-2024-05-13',
'gpt-4o-2024-08-06', 'gpt-4o-2024-08-06',
'chatgpt-4o-latest',
'gpt-4-turbo-2024-04-09', 'gpt-4-turbo-2024-04-09',
'gpt-4o-mini-2024-07-18']: 'gpt-4o-mini-2024-07-18']:
from .gpt import generate_response from .gpt import generate_response
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment