@@ -75,7 +75,7 @@ class GptParams:
75
75
76
76
def gpt_params_parse (argv = None ):
77
77
parser = argparse .ArgumentParser (
78
- formatter_class = argparse .ArgumentDefaultsHelpFormatter
78
+ formatter_class = argparse .ArgumentDefaultsHelpFormatter ,
79
79
)
80
80
parser .add_argument (
81
81
"-s" ,
@@ -102,7 +102,7 @@ def gpt_params_parse(argv=None):
102
102
dest = "n_predict" ,
103
103
)
104
104
parser .add_argument (
105
- "--n_parts" , type = int , default = - 1 , help = "number of model parts" , dest = "n_parts"
105
+ "--n_parts" , type = int , default = - 1 , help = "number of model parts" , dest = "n_parts" ,
106
106
)
107
107
parser .add_argument (
108
108
"-c" ,
@@ -143,10 +143,10 @@ def gpt_params_parse(argv=None):
143
143
dest = "ignore_eos" ,
144
144
)
145
145
parser .add_argument (
146
- "--top_k" , type = int , default = 40 , help = "top-k sampling" , dest = "top_k"
146
+ "--top_k" , type = int , default = 40 , help = "top-k sampling" , dest = "top_k" ,
147
147
)
148
148
parser .add_argument (
149
- "--top_p" , type = float , default = 0.95 , help = "top-p samplin" , dest = "top_p"
149
+ "--top_p" , type = float , default = 0.95 , help = "top-p samplin" , dest = "top_p" ,
150
150
)
151
151
parser .add_argument (
152
152
"--tfs" ,
@@ -156,7 +156,7 @@ def gpt_params_parse(argv=None):
156
156
dest = "tfs_z" ,
157
157
)
158
158
parser .add_argument (
159
- "--temp" , type = float , default = 0.80 , help = "temperature" , dest = "temp"
159
+ "--temp" , type = float , default = 0.80 , help = "temperature" , dest = "temp" ,
160
160
)
161
161
parser .add_argument (
162
162
"--repeat_penalty" ,
@@ -217,7 +217,7 @@ def gpt_params_parse(argv=None):
217
217
dest = "model" ,
218
218
)
219
219
parser .add_argument (
220
- "-p" , "--prompt" , type = str , default = None , help = "initial prompt" , dest = "prompt"
220
+ "-p" , "--prompt" , type = str , default = None , help = "initial prompt" , dest = "prompt" ,
221
221
)
222
222
parser .add_argument (
223
223
"-f" ,
@@ -242,7 +242,7 @@ def gpt_params_parse(argv=None):
242
242
dest = "input_prefix" ,
243
243
)
244
244
parser .add_argument (
245
- "--in-suffix" , type = str , default = "" , help = "append to input" , dest = "input_suffix"
245
+ "--in-suffix" , type = str , default = "" , help = "append to input" , dest = "input_suffix" ,
246
246
)
247
247
parser .add_argument (
248
248
"-r" ,
0 commit comments