北念
2023-10-17 72a0600129fba948fc7eb4ddb187404a6e0a2db3
fix bug in whisper inference
1个文件已修改
8 ■■■■ 已修改文件
funasr/bin/asr_inference_launch.py 8 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/asr_inference_launch.py
@@ -2056,8 +2056,12 @@
    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: