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

fix bug in whisper inference

Browse Source
This commit is contained in:
北念 2023-10-17 14:15:14 +08:00
parent fde48a8652
commit 72a0600129
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
funasr/bin/asr_inference_launch.py
View File

@ -2056,8 +2056,12 @@ def inference_whisper(
ncpu = kwargs.get("ncpu", 1)
torch.set_num_threads(ncpu)
language = param_dict.get("language", None)
task = param_dict.get("task", "transcribe")
if param_dict:
language = param_dict.get("language", None)
task = param_dict.get("task", "transcribe")
else:
language = None
task = "transcribe"
if batch_size > 1:
raise NotImplementedError("batch decoding is not implemented")
if word_lm_train_config is not None: