funasr/bin/vad_inference_online.py
@@ -151,6 +151,9 @@ **kwargs, ): assert check_argument_types() ncpu = kwargs.get("ncpu", 1) torch.set_num_threads(ncpu) if batch_size > 1: raise NotImplementedError("batch decoding is not implemented") if ngpu > 1: @@ -240,7 +243,6 @@ item = {'key': keys[i], 'value': results[i]} vad_results.append(item) if writer is not None: results[i] = json.loads(results[i]) ibest_writer["text"][keys[i]] = "{}".format(results[i]) return vad_results