[feature-fix] adding gpt-4o option to arguments

Signed-off-by: ltn18 <nguyentamhoanglam@gmail.com>
This commit is contained in:
ltn18 2024-10-09 23:23:01 -04:00
parent b441123518
commit 334fef0a8f

2
run.py
View File

@ -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'])