From 334fef0a8fdd63ffcb83f7506132253cff37bce5 Mon Sep 17 00:00:00 2001 From: ltn18 Date: Wed, 9 Oct 2024 23:23:01 -0400 Subject: [PATCH] [feature-fix] adding gpt-4o option to arguments Signed-off-by: ltn18 --- run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run.py b/run.py index e36712a..e4b6777 100644 --- a/run.py +++ b/run.py @@ -42,7 +42,7 @@ def run(args): def parse_args(): args = argparse.ArgumentParser() - args.add_argument('--backend', type=str, choices=['gpt-4', 'gpt-3.5-turbo'], default='gpt-4') + args.add_argument('--backend', type=str, choices=['gpt-4', 'gpt-3.5-turbo', 'gpt-4o'], default='gpt-4') args.add_argument('--temperature', type=float, default=0.7) args.add_argument('--task', type=str, required=True, choices=['game24', 'text', 'crosswords'])