funasr/build_utils/build_model.py
@@ -13,6 +13,8 @@ model = build_lm_model(args) elif args.task_name == "punc": model = build_punc_model(args) elif args.task_name == "vad": model = build_vad_model(args) else: raise NotImplementedError("Not supported task: {}".format(args.task_name))