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