From babfab2d5e2305982e0a436ab8ac3aaf109e91d6 Mon Sep 17 00:00:00 2001 From: Chao Gao Date: Sun, 22 Oct 2023 20:18:25 +0800 Subject: [PATCH] Make voice file extension comparison case-insensitive --- tortoise/utils/audio.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tortoise/utils/audio.py b/tortoise/utils/audio.py index 6842af5..e263435 100644 --- a/tortoise/utils/audio.py +++ b/tortoise/utils/audio.py @@ -27,9 +27,10 @@ def load_wav_to_torch(full_path): def load_audio(audiopath, sampling_rate): - if audiopath[-4:] == '.wav': + extension = os.path.splitext(audiopath)[1].casefold() + if extension == '.wav': audio, lsr = load_wav_to_torch(audiopath) - elif audiopath[-4:] == '.mp3': + elif extension == '.mp3': audio, lsr = librosa.load(audiopath, sr=sampling_rate) audio = torch.FloatTensor(audio) else: