funasr/build_utils/build_asr_model.py
@@ -408,10 +408,15 @@ **args.model_conf, ) elif args.model == "timestamp_prediction": # predictor predictor_class = predictor_choices.get_class(args.predictor) predictor = predictor_class(**args.predictor_conf) model_class = model_choices.get_class(args.model) model = model_class( frontend=frontend, encoder=encoder, predictor=predictor, token_list=token_list, **args.model_conf, )
