Merge pull request #49 from BushyToaster88/fix-torch-stft-error-on-gp…

…us-sm-53 fix-torch-stft-error-on-gpus-sm-53
Plachtaa · Mar 3, 2023 · da3670a · da3670a
2 parents 8137fb9 + b92361b
commit da3670a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mel_processing.py b/mel_processing.py
@@ -101,8 +101,8 @@ def mel_spectrogram_torch(y, n_fft, num_mels, sampling_rate, hop_size, win_size,
     y = torch.nn.functional.pad(y.unsqueeze(1), (int((n_fft-hop_size)/2), int((n_fft-hop_size)/2)), mode='reflect')
     y = y.squeeze(1)
 
-    spec = torch.stft(y, n_fft, hop_length=hop_size, win_length=win_size, window=hann_window[wnsize_dtype_device],
-                      center=center, pad_mode='reflect', normalized=False, onesided=True)
+    spec = torch.stft(y.float(), n_fft, hop_length=hop_size, win_length=win_size, window=hann_window[wnsize_dtype_device],
+        center=center, pad_mode='reflect', normalized=False, onesided=True)
 
     spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6)