Merge pull request #2 from ilaria-manco/master

KentoNishi · Mar 29, 2022 · 6ac2a1d · 6ac2a1d
2 parents cf9a707 + 1e534ec
commit 6ac2a1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_pitch_shift/main.py b/torch_pitch_shift/main.py
@@ -149,7 +149,7 @@ def pitch_shift(
     resampler = T.Resample(sample_rate, int(sample_rate / shift)).to(input.device)
     output = input
     output = output.reshape(batch_size * channels, samples)
-    output = torch.stft(output, n_fft, hop_length)[None, ...]
+    output = torch.stft(output, n_fft, hop_length, return_complex=True)[None, ...]
     stretcher = T.TimeStretch(
         fixed_rate=float(1 / shift), n_freq=output.shape[2], hop_length=hop_length
     ).to(input.device)