Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixup

This commit is contained in:
Eren Gölge 2021-12-13 16:30:15 +00:00
parent 3818bd0c23
commit 3c6d7f495c
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/speaker_encoder/models/resnet.py
View File

@ -1,12 +1,11 @@
import numpy as np import numpy as np
import torch import torch
import torchaudio
from torch import nn from torch import nn
from TTS.utils.audio import TorchSTFT from TTS.utils.audio import TorchSTFT
from TTS.utils.io import load_fsspec from TTS.utils.io import load_fsspec
import torchaudio
class PreEmphasis(nn.Module): class PreEmphasis(nn.Module):
def __init__(self, coefficient=0.97): def __init__(self, coefficient=0.97):
@ -134,7 +133,7 @@ class ResNetSpeakerEncoder(nn.Module):
hop_length=audio_config["hop_length"], hop_length=audio_config["hop_length"],
window_fn=torch.hamming_window, window_fn=torch.hamming_window,
n_mels=audio_config["num_mels"], n_mels=audio_config["num_mels"],
) ),
) )
else: else:
self.torch_spec = None self.torch_spec = None