shixian.shi
2024-01-12 09a28d19df5854bdd4bd4d3a05dcb6f502ec6b07
update
1个文件已修改
1 ■■■■ 已修改文件
funasr/bin/inference.py 1 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/inference.py
@@ -168,7 +168,6 @@
            device = "cpu"
            # kwargs["batch_size"] = 1
        kwargs["device"] = device
        import pdb; pdb.set_trace()
        
        if kwargs.get("ncpu", None):
            torch.set_num_threads(kwargs.get("ncpu"))