From df204f4f8bdaf998fa86660fc2a8687ae9bb7433 Mon Sep 17 00:00:00 2001 From: Ftps Date: Sun, 10 Dec 2023 16:27:58 +0900 Subject: [PATCH] fix onnx flow --- rvc/lib/infer_pack/models_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rvc/lib/infer_pack/models_onnx.py b/rvc/lib/infer_pack/models_onnx.py index 094bf1a..81a533c 100644 --- 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):