Commit b3cd8a47 authored by zihanl's avatar zihanl
Browse files

add api argument

parent 48dbd9ca

tasks/main.py

+6 −4
Original line number Diff line number Diff line
@@ -90,20 +90,22 @@ def get_tasks_args(parser): 
                       'each line is an input.')

    group.add_argument("--sample-output-file", type=str, default=None,

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

    group.add_argument('--prompt-type', type=str, default="",

    group.add_argument('--prompt-type', type=str, default=None,

                       help='prompt type (knowledge or response)')

    group.add_argument('--num-prompt-examples', type=int, default=10,

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

    group.add_argument('--answer-file', type=str, default="",

    group.add_argument('--answer-file', type=str, default=None,

                       help='datapath for golden sentences')

    group.add_argument('--out-seq-length', type=int, default=100,

                       help='output sequence length')

    group.add_argument('--api-prompt', default=False, action="store_true",

                       help='setup model api for prompting')

    group.add_argument('--megatron-api-url', type=str, default=None,

                       help='url of the megatron api')



    return parser