From d17d48935a3d09dfaebe182ee3bb11795f06a85c Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期一, 24 四月 2023 16:41:43 +0800
Subject: [PATCH] update

---
 funasr/build_utils/build_args.py |    4 ++--
 1 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/funasr/build_utils/build_args.py b/funasr/build_utils/build_args.py
index b77cbcc..1d0728b 100644
--- a/funasr/build_utils/build_args.py
+++ b/funasr/build_utils/build_args.py
@@ -8,7 +8,7 @@
 from funasr.utils.types import str_or_none
 
 
-def build_args(args):
+def build_args(args, extra_task_params):
     parser = argparse.ArgumentParser("Task related config")
     if args.task_name == "asr":
         from funasr.build_utils.build_asr_model import class_choices_list
@@ -85,5 +85,5 @@
     else:
         raise NotImplementedError("Not supported task: {}".format(args.task_name))
 
-    task_args = parser.parse_args()
+    task_args = parser.parse_args(extra_task_params)
     return task_args

--
Gitblit v1.9.1