Skip to content

GitLab

Commit b3cd8a47 authored by zihanl's avatar zihanl
Browse files

add api argument

parent 48dbd9ca
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 4 deletions
tasks/main.py
View file @ b3cd8a47
...@@ -90,20 +90,22 @@ def get_tasks_args(parser): ...@@ -90,20 +90,22 @@ def get_tasks_args(parser):
'each line is an input.') 'each line is an input.')
group.add_argument("--sample-output-file", type=str, default=None, group.add_argument("--sample-output-file", type=str, default=None,
help='Output file got from --sample-input-file') help='Output file got from --sample-input-file')
group.add_argument('--prompt-file', type=str, default="", group.add_argument('--prompt-file', type=str, default=None,
help='prompting file') help='prompting file')
group.add_argument('--prompt-type', type=str, default="", group.add_argument('--prompt-type', type=str, default=None,
help='prompt type (knowledge or response)') help='prompt type (knowledge or response)')
group.add_argument('--num-prompt-examples', type=int, default=10, group.add_argument('--num-prompt-examples', type=int, default=10,
help='number of prompt examples') help='number of prompt examples')
group.add_argument('--guess-file', type=str, default="", group.add_argument('--guess-file', type=str, default=None,
help='datapath for generated sentences') help='datapath for generated sentences')
group.add_argument('--answer-file', type=str, default="", group.add_argument('--answer-file', type=str, default=None,
help='datapath for golden sentences') help='datapath for golden sentences')
group.add_argument('--out-seq-length', type=int, default=100, group.add_argument('--out-seq-length', type=int, default=100,
help='output sequence length') help='output sequence length')
group.add_argument('--api-prompt', default=False, action="store_true", group.add_argument('--api-prompt', default=False, action="store_true",
help='setup model api for prompting') help='setup model api for prompting')
group.add_argument('--megatron-api-url', type=str, default=None,
help='url of the megatron api')
return parser return parser
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment