From 15f362d5b108db0955e46184ad6d073efb09983e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Thu, 8 Apr 2021 11:13:55 +0200 Subject: [PATCH] formatting --- TTS/vocoder/layers/losses.py | 1 + TTS/vocoder/models/hifigan_discriminator.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) 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