Merge pull request #389 from alibaba-damo-academy/main

yhliang

2023-04-20 5de9e75d587b752d8d1063cc7903c4571df99189

 funasr/bin/lm_inference.py

@@ -89,10 +89,9 @@
    **kwargs,
):
    assert check_argument_types()
    logging.basicConfig(
        level=log_level,
        format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s",
    )
    ncpu = kwargs.get("ncpu", 1)
    torch.set_num_threads(ncpu)


    if ngpu >= 1 and torch.cuda.is_available():
        device = "cuda"