zhifu gao
2024-03-11 e847f85a14b85a94cfa6e109ecab69974eeeb443
funasr/auto/auto_model.py
@@ -156,8 +156,8 @@
            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)