This commit is contained in:
Eren Gölge 2021-12-13 16:30:15 +00:00
parent 6274d5e438
commit 9ec6238f4a
1 changed files with 2 additions and 3 deletions

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