-
Notifications
You must be signed in to change notification settings - Fork 17
/
config.py
20 lines (13 loc) · 668 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import argparse
def parse_arguments():
parser = argparse.ArgumentParser()
parser.add_argument("--data_path", type=str, help="")
parser.add_argument("--temperature", type=float, default=0.7)
parser.add_argument("--prompt_name", type=str, help="")
parser.add_argument("--character", type=str, help="", default="Beethoven")
parser.add_argument("--model_name", type=str, help="", default="gpt-3.5-turbo")
parser.add_argument("--debug", type=int, default=0, help="")
parser.add_argument("--max_tokens", type=int, default=None, help="")
parsed_args = parser.parse_args()
return parsed_args
args = parse_arguments()