北念
2023-10-17 fde48a865253b21f874dedf384c1bd8b59481112
funasr/bin/asr_inference_launch.py
@@ -2056,6 +2056,8 @@
    ncpu = kwargs.get("ncpu", 1)
    torch.set_num_threads(ncpu)
    language = param_dict.get("language", None)
    task = param_dict.get("task", "transcribe")
    if batch_size > 1:
        raise NotImplementedError("batch decoding is not implemented")
    if word_lm_train_config is not None:
@@ -2099,6 +2101,8 @@
        penalty=penalty,
        nbest=nbest,
        streaming=streaming,
        language=language,
        task=task,
    )
    logging.info("speech2text_kwargs: {}".format(speech2text_kwargs))
    speech2text = Speech2TextWhisper(**speech2text_kwargs)