From 48e4cbd6e9d23b8b62e424bd55fd94a1da689e32 Mon Sep 17 00:00:00 2001 From: Trung Le Date: Thu, 13 Apr 2023 17:36:01 +0700 Subject: [PATCH] Correcting librosa mel compute --- src/python/piper_train/vits/mel_processing.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/python/piper_train/vits/mel_processing.py b/src/python/piper_train/vits/mel_processing.py index 1776f50..72e81e2 100644 --- a/src/python/piper_train/vits/mel_processing.py +++ b/src/python/piper_train/vits/mel_processing.py @@ -103,7 +103,9 @@ def mel_spectrogram_torch( fmax_dtype_device = str(fmax) + "_" + dtype_device wnsize_dtype_device = str(win_size) + "_" + dtype_device if fmax_dtype_device not in mel_basis: - mel = librosa_mel_fn(sampling_rate, n_fft, num_mels, fmin, fmax) + mel = librosa_mel_fn( + sr=sampling_rate, n_fft=n_fft, n_mels=num_mels, fmin=fmin, fmax=fmax + ) mel_basis[fmax_dtype_device] = torch.from_numpy(mel).type_as(y) if wnsize_dtype_device not in hann_window: hann_window[wnsize_dtype_device] = torch.hann_window(win_size).type_as(y)