This commit is contained in:
Eren Gölge
2021-12-20 11:54:10 +00:00
parent 3818bd0c23
commit 3c6d7f495c
+2 -3
View File
@@ -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