fix flow (#1605)
This commit is contained in:
parent
7c753e6ac1
commit
786005f0de
@ -149,7 +149,7 @@ class ResidualCouplingBlock(nn.Module):
|
|||||||
x, _ = flow(x, x_mask, g=g, reverse=reverse)
|
x, _ = flow(x, x_mask, g=g, reverse=reverse)
|
||||||
else:
|
else:
|
||||||
for flow in reversed(self.flows):
|
for flow in reversed(self.flows):
|
||||||
x = flow(x, x_mask, g=g, reverse=reverse)
|
x, _ = flow(x, x_mask, g=g, reverse=reverse)
|
||||||
return x
|
return x
|
||||||
|
|
||||||
def remove_weight_norm(self):
|
def remove_weight_norm(self):
|
||||||
|
Loading…
Reference in New Issue
Block a user