| funasr/auto/auto_model.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
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)