mirror of
https://github.com/neonbjb/tortoise-tts.git
synced 2026-04-21 06:14:04 +00:00
Make voice file extension comparison case-insensitive
This commit is contained in:
parent
9a954327c7
commit
babfab2d5e
1 changed files with 3 additions and 2 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue