funasr/utils/vad_utils.py
@@ -1,6 +1,7 @@ import torch from torch.nn.utils.rnn import pad_sequence def slice_padding_fbank(speech, speech_lengths, vad_segments): speech_list = [] speech_lengths_list = [] @@ -15,7 +16,6 @@ feats_pad = pad_sequence(speech_list, batch_first=True, padding_value=0.0) speech_lengths_pad = torch.Tensor(speech_lengths_list).int() return feats_pad, speech_lengths_pad def slice_padding_audio_samples(speech, speech_lengths, vad_segments): speech_list = []