neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-03-21 04:34:41 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make voice file extension comparison case-insensitive

Browse source
This commit is contained in:
Chao Gao 2023-10-22 20:18:25 +08:00 committed by GitHub
parent 9a954327c7
commit babfab2d5e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tortoise/utils/audio.py
View file

@ -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: