diff --git a/TTS/vocoder/layers/losses.py b/TTS/vocoder/layers/losses.py index e1d27c24..268e6096 100644 --- a/TTS/vocoder/layers/losses.py +++ b/TTS/vocoder/layers/losses.py @@ -33,6 +33,7 @@ class TorchSTFT(nn.Module): # pylint: disable=abstract-method self.mel_basis = None if use_mel: self._build_mel_basis() + @torch.no_grad() def __call__(self, x): """Compute spectrogram frames by torch based stft. diff --git a/TTS/vocoder/models/hifigan_discriminator.py b/TTS/vocoder/models/hifigan_discriminator.py index 5e508a99..d506a56b 100644 --- a/TTS/vocoder/models/hifigan_discriminator.py +++ b/TTS/vocoder/models/hifigan_discriminator.py @@ -1,5 +1,4 @@ # adopted from https://github.com/jik876/hifi-gan/blob/master/models.py - import torch from torch import nn from torch.nn import functional as F