zhifu gao
2024-03-11 e847f85a14b85a94cfa6e109ecab69974eeeb443
funasr/auto/auto_model.py
@@ -155,9 +155,9 @@
            device = "cpu"
            kwargs["batch_size"] = 1
        kwargs["device"] = device
        if kwargs.get("ncpu", 4):
            torch.set_num_threads(kwargs.get("ncpu"))
        torch.set_num_threads(kwargs.get("ncpu", 4))
        
        # build tokenizer
        tokenizer = kwargs.get("tokenizer", None)