funasr/build_utils/build_args.py
@@ -86,6 +86,12 @@ from funasr.build_utils.build_diar_model import class_choices_list for class_choices in class_choices_list: class_choices.add_arguments(task_parser) task_parser.add_argument( "--input_size", type=int_or_none, default=None, help="The number of input dimension of the feature", ) elif args.task_name == "sv": from funasr.build_utils.build_sv_model import class_choices_list