restore to original main.py

                        help='Av.rank validation: how many other negatives to'

                        ' take from each question pool')

    # parameters for the knowledgeable dialogue generation

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

                       help='Get input from file instead of interactive mode, '

                       '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=None,

                       help='prompting file')

    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=None,

                       help='datapath for generated sentences')

    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

        from orqa.evaluate_orqa import main

    elif args.task in ['RET-FINETUNE-NQ']:

        from orqa.supervised.finetune import main

    elif args.task == 'KNWL-DIALO-PROMPT':

        from knwl_dialo.prompt import main

    elif args.task == 'KNWL-DIALO-EVAL-F1':

        from knwl_dialo.evaluate import main

    else:

        raise NotImplementedError('Task {} is not implemented.'.format(

            args.task))