zhifu gao
2023-05-05 5ed3c73b78dcd441b9fb7f9b8aa509a0648dbac7
funasr/bin/asr_inference.py
@@ -346,6 +346,8 @@
    **kwargs,
):
    assert check_argument_types()
    ncpu = kwargs.get("ncpu", 1)
    torch.set_num_threads(ncpu)
    if batch_size > 1:
        raise NotImplementedError("batch decoding is not implemented")
    if word_lm_train_config is not None: