shixian.shi
2024-01-12 09a28d19df5854bdd4bd4d3a05dcb6f502ec6b07
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"))