1
0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-08 20:10:44 +08:00

optimize(infer): move transforms into rvc

This commit is contained in:
源文雨
2024-06-07 19:53:23 +09:00
parent 49488dcae9
commit 978abd8aac
2 changed files with 21 additions and 21 deletions

View File

@@ -12,7 +12,7 @@ from torch.nn.utils import remove_weight_norm, weight_norm
from rvc import utils
from rvc.utils import get_padding, call_weight_data_normal_if_Conv
from infer.lib.infer_pack.transforms import piecewise_rational_quadratic_transform
from rvc.transforms import piecewise_rational_quadratic_transform
LRELU_SLOPE = 0.1
@@ -583,7 +583,7 @@ class ConvFlow(nn.Module):
reverse=False,
):
x0, x1 = torch.split(x, [self.half_channels] * 2, 1)
h = self.pre(x0)
h: torch.Tensor = self.pre(x0)
h = self.convs(h, x_mask, g=g)
h = self.proj(h) * x_mask