modelscope/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR.git synced 2026-10-12 02:19:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

update tp_inference

Browse Source
This commit is contained in:
shixian.shi
2023-03-10 10:50:28 +08:00
parent ed2370591f
commit ca98012cd2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
funasr/bin/tp_inference.py
+2 -2
View File
@@ -263,7 +263,7 @@ def inference_modelscope(
preprocessor = LMPreprocessor(
train=False,
token_type=speechtext2timestamp.tp_train_args.token_type,
token_list=speechtext2timestamp.tp_train_args,
token_list=speechtext2timestamp.tp_train_args.token_list,
bpemodel=None,
text_cleaner=None,
g2p_type=None,
@@ -293,7 +293,7 @@ def inference_modelscope(
batch_size=batch_size,
key_file=key_file,
num_workers=num_workers,
preprocess_fn=LMPreprocessor,
preprocess_fn=preprocessor,
collate_fn=ASRTask.build_collate_fn(speechtext2timestamp.tp_train_args, False),
allow_variable_data_keys=allow_variable_data_keys,
inference=True,