Skip to content

Commit ff353bc

Browse files
authored
Lint
1 parent 9d8ecd1 commit ff353bc

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

examples/low_level_api/common.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ class GptParams:
7575

7676
def gpt_params_parse(argv=None):
7777
parser = argparse.ArgumentParser(
78-
formatter_class=argparse.ArgumentDefaultsHelpFormatter
78+
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
7979
)
8080
parser.add_argument(
8181
"-s",
@@ -102,7 +102,7 @@ def gpt_params_parse(argv=None):
102102
dest="n_predict",
103103
)
104104
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",
106106
)
107107
parser.add_argument(
108108
"-c",
@@ -143,10 +143,10 @@ def gpt_params_parse(argv=None):
143143
dest="ignore_eos",
144144
)
145145
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",
147147
)
148148
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",
150150
)
151151
parser.add_argument(
152152
"--tfs",
@@ -156,7 +156,7 @@ def gpt_params_parse(argv=None):
156156
dest="tfs_z",
157157
)
158158
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",
160160
)
161161
parser.add_argument(
162162
"--repeat_penalty",
@@ -217,7 +217,7 @@ def gpt_params_parse(argv=None):
217217
dest="model",
218218
)
219219
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",
221221
)
222222
parser.add_argument(
223223
"-f",
@@ -242,7 +242,7 @@ def gpt_params_parse(argv=None):
242242
dest="input_prefix",
243243
)
244244
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",
246246
)
247247
parser.add_argument(
248248
"-r",

0 commit comments

Comments
 (0)