1
0
mirror of synced 2024-11-27 17:00:54 +01:00
This commit is contained in:
Ftps 2023-12-10 16:32:44 +09:00 committed by GitHub
parent 7c753e6ac1
commit 786005f0de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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):