fix onnx flow

2023-12-10 16:27:58 +09:00
parent db6c80b618
commit df204f4f8b
1 changed files with 1 additions and 1 deletions
--- a/rvc/lib/infer_pack/models_onnx.py
+++ b/rvc/lib/infer_pack/models_onnx.py
@@ -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):