This commit is contained in:
Ftps
2023-12-10 16:32:44 +09:00
committed by GitHub
parent 7c753e6ac1
commit 786005f0de

View File

@@ -149,7 +149,7 @@ class ResidualCouplingBlock(nn.Module):
x, _ = flow(x, x_mask, g=g, reverse=reverse)
else:
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
def remove_weight_norm(self):