diff --git a/tasks/main.py b/tasks/main.py index 574e29f8d9..6d8fc8f5fd 100644 --- a/tasks/main.py +++ b/tasks/main.py @@ -84,28 +84,6 @@ def get_tasks_args(parser): 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 @@ -130,10 +108,6 @@ def get_tasks_args(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))