funasr/bin/vad_inference_launch.py
@@ -90,6 +90,7 @@ preprocess_args=None, data_path_and_name_and_type=data_path_and_name_and_type, dtype=dtype, fs=fs, batch_size=batch_size, key_file=key_file, num_workers=num_workers, @@ -123,7 +124,7 @@ vad_results.append(item) if writer is not None: ibest_writer["text"][keys[i]] = "{}".format(results[i]) torch.cuda.empty_cache() return vad_results return _forward @@ -189,6 +190,7 @@ preprocess_args=None, data_path_and_name_and_type=data_path_and_name_and_type, dtype=dtype, fs=fs, batch_size=batch_size, key_file=key_file, num_workers=num_workers,