diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py index 28b9e94fc..c8cd30c0c 100644 --- a/funasr/auto/auto_model.py +++ b/funasr/auto/auto_model.py @@ -155,9 +155,9 @@ class AutoModel: 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)