v0.4.6

游雁

2023-05-08 63f1496cbb9d1cf5893a77489a9c8e5ef589ccf0

 funasr/bin/punctuation_infer_vadrealtime.py

@@ -203,10 +203,8 @@
    **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"