From 9ec6238f4ad08d502ed37a27089c8e63d2ff5fc3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Mon, 13 Dec 2021 16:30:15 +0000 Subject: [PATCH] Fixup --- TTS/speaker_encoder/models/resnet.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/TTS/speaker_encoder/models/resnet.py b/TTS/speaker_encoder/models/resnet.py index 3b96f270..92d34494 100644 --- a/TTS/speaker_encoder/models/resnet.py +++ b/TTS/speaker_encoder/models/resnet.py @@ -1,12 +1,11 @@ import numpy as np import torch +import torchaudio from torch import nn from TTS.utils.audio import TorchSTFT from TTS.utils.io import load_fsspec -import torchaudio - class PreEmphasis(nn.Module): def __init__(self, coefficient=0.97): @@ -134,7 +133,7 @@ class ResNetSpeakerEncoder(nn.Module): hop_length=audio_config["hop_length"], window_fn=torch.hamming_window, n_mels=audio_config["num_mels"], - ) + ), ) else: self.torch_spec = None