speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
speech_asr 2023-04-20 16:03:54 +08:00
parent 200d1ede05
commit a29166b9a0
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
funasr/utils/build_asr_model.py
View File

@ -268,7 +268,7 @@ def build_asr_model(args):
token_list=token_list,
**args.model_conf,
)
elif args.model == "paraformer":
elif args.model in ["paraformer", "paraformer_bert", "bicif_paraformer", "contextual_paraformer"]:
# predictor
predictor_class = predictor_choices.get_class(args.predictor)
predictor = predictor_class(**args.predictor_conf)
@ -336,7 +336,14 @@ def build_asr_model(args):
stride_conv=stride_conv,
**args.model_conf,
)
elif args.model == "timestamp_prediction":
model_class = model_choices.get_class(args.model)
model = model_class(
frontend=frontend,
encoder=encoder,
token_list=token_list,
**args.model_conf,
)
else:
raise NotImplementedError("Not supported model: {}".format(args.model))