Add error message

This commit is contained in:
James Betker 2022-05-12 20:15:40 -06:00
parent c4a5a23985
commit f5ebd14d09

View File

@ -29,6 +29,8 @@ def load_audio(audiopath, sampling_rate):
elif audiopath[-4:] == '.mp3':
audio, lsr = librosa.load(audiopath, sr=sampling_rate)
audio = torch.FloatTensor(audio)
else:
assert False, f"Unsupported audio format provided: {audiopath[-4:]}"
# Remove any channel data.
if len(audio.shape) > 1: