mirror of
https://github.com/neonbjb/tortoise-tts.git
synced 2026-01-05 00:00:49 +01:00
Merge pull request #650 from reterVision/patch-1
Make voice file extension comparison case-insensitive
This commit is contained in:
commit
cce72b2222
|
|
@ -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…
Reference in a new issue