Spaces:
Running
on
A10G
Running
on
A10G
Update audioldm/audio/stft.py
Browse files- audioldm/audio/stft.py +1 -1
audioldm/audio/stft.py
CHANGED
@@ -148,7 +148,7 @@ class TacotronSTFT(torch.nn.Module):
|
|
148 |
self.n_mel_channels = n_mel_channels
|
149 |
self.sampling_rate = sampling_rate
|
150 |
self.stft_fn = STFT(filter_length, hop_length, win_length)
|
151 |
-
mel_basis =
|
152 |
sr=sampling_rate,
|
153 |
n_fft=filter_length,
|
154 |
n_mels=n_mel_channels,
|
|
|
148 |
self.n_mel_channels = n_mel_channels
|
149 |
self.sampling_rate = sampling_rate
|
150 |
self.stft_fn = STFT(filter_length, hop_length, win_length)
|
151 |
+
mel_basis = librosa_mel_fn(
|
152 |
sr=sampling_rate,
|
153 |
n_fft=filter_length,
|
154 |
n_mels=n_mel_channels,
|