Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-24 14:35:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add error handling to log_mel_spectrogram function

Browse Source
This commit is contained in:
Madhumitha 2024-05-04 04:57:53 +05:30
parent ba3f3cd54b
commit 49c833dcee
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
whisper/audio.py
View File

@ -140,6 +140,9 @@ def log_mel_spectrogram(
audio = load_audio(audio)
audio = torch.from_numpy(audio)
if len(audio) < N_SAMPLES: # Check if audio length is less than expected
raise ValueError("Input audio length is shorter than the expected length.")
if device is not None:
audio = audio.to(device)
if padding > 0: