update device

shixian.shi

2024-01-12 c3442d9566f5a2011c95b0d2998958a1b5348564

 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"))