Update audio_processor.py

fixed out range bug mentioned in issues/297
This commit is contained in:
Octpus
2025-04-08 15:51:06 +08:00
committed by GitHub
parent 7b829ba778
commit 36163fccbd

View File

@@ -64,7 +64,7 @@ class AudioProcessor:
whisper_feature = whisper_feature[:,:actual_length,...] whisper_feature = whisper_feature[:,:actual_length,...]
# Calculate padding amount # Calculate padding amount
padding_nums = math.floor(whisper_idx_multiplier) padding_nums = math.ceil(whisper_idx_multiplier)
# Add padding at start and end # Add padding at start and end
whisper_feature = torch.cat([ whisper_feature = torch.cat([
torch.zeros_like(whisper_feature[:, :padding_nums * audio_padding_length_left]), torch.zeros_like(whisper_feature[:, :padding_nums * audio_padding_length_left]),