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

revert some of Hifigan generator updates

pull/422/head
Eren Gölge 2021-04-07 19:15:31 +02:00
parent 02bc776c35
commit 13dca6e6b6
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
TTS/vocoder/models/hifigan_generator.py
View File

@ -85,12 +85,12 @@ class ResBlock1(torch.nn.Module):
x: [B, C, T] x: [B, C, T]
""" """
for c1, c2 in zip(self.convs1, self.convs2): for c1, c2 in zip(self.convs1, self.convs2):
o = F.leaky_relu(x, LRELU_SLOPE) xt = F.leaky_relu(x, LRELU_SLOPE)
o = c1(o) xt = c1(xt)
o = F.leaky_relu(o, LRELU_SLOPE) xt = F.leaky_relu(xt, LRELU_SLOPE)
o = c2(o) xt = c2(xt)
o = o + x x = xt + x
return o return x
def remove_weight_norm(self): def remove_weight_norm(self):
for l in self.convs1: for l in self.convs1:
@ -134,10 +134,10 @@ class ResBlock2(torch.nn.Module):
def forward(self, x): def forward(self, x):
for c in self.convs: for c in self.convs:
o = F.leaky_relu(x, LRELU_SLOPE) xt = F.leaky_relu(x, LRELU_SLOPE)
o = c(o) xt = c(xt)
o = o + x x = xt + x
return o return x
def remove_weight_norm(self): def remove_weight_norm(self):
for l in self.convs: for l in self.convs:
@ -223,7 +223,6 @@ class HifiganGenerator(torch.nn.Module):
o = F.leaky_relu(o) o = F.leaky_relu(o)
o = self.conv_post(o) o = self.conv_post(o)
o = torch.tanh(o) o = torch.tanh(o)
breakpoint()
return o return o
@torch.no_grad() @torch.no_grad()