remove finetune part

5f4e63fc · zihanl · 18de984a · 5f4e63fc
Commit 5f4e63fc authored Dec 06, 2021 by zihanl
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 2 deletions

tasks/main.py tasks/main.py +0 -2

No files found.
--- a/tasks/main.py
+++ b/tasks/main.py
@@ -134,8 +134,6 @@ if __name__ == '__main__':
        from orqa.supervised.finetune import main
    elif args.task == 'KNWL-DIALO-PROMPT':
        from knwl_dialo.prompt import main
-    elif args.task in ['KNWL-DIALO-FINETUNE', 'KNWL-DIALO-GEN']:
-        from knwl_dialo.finetune import main
    elif args.task == 'KNWL-DIALO-EVAL-F1':
        from knwl_dialo.evaluate import main
    else: