fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-06 17:50:25 +08:00
Code Activity

fix(rtrvc): skip head unimplemented

Browse Source
This commit is contained in:
源文雨
2024-06-16 16:46:59 +09:00
parent df83554ac1
commit 0d5cd347bc
5 changed files with 32 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
rvc/layers/generators.py
View File

@@ -61,23 +61,21 @@ class Generator(torch.nn.Module):
self,
x: torch.Tensor,
g: Optional[torch.Tensor] = None,
# n_res: Optional[torch.Tensor] = None,
n_res: Optional[int] = None,
) -> torch.Tensor:
return super().__call__(x, g=g)
return super().__call__(x, g=g, n_res=n_res)
def forward(
self,
x: torch.Tensor,
g: Optional[torch.Tensor] = None,
# n_res: Optional[torch.Tensor] = None,
n_res: Optional[int] = None,
):
"""
if n_res is not None:
assert isinstance(n_res, torch.Tensor)
n = int(n_res.item())
n = int(n_res)
if n != x.shape[-1]:
x = F.interpolate(x, size=n, mode="linear")
"""
x = self.conv_pre(x)
if g is not None:
x = x + self.cond(g)