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-07 19:40:44 +08:00
Code Activity

fix(train): mysterious importing order

Browse Source
This commit is contained in:
源文雨
2024-06-16 19:33:23 +09:00
parent add4642b7e
commit d9a116f4f7
3 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rvc/layers/encoders.py
View File

@@ -192,7 +192,7 @@ class PosteriorEncoder(nn.Module):
def __call__(
self, x: torch.Tensor, x_lengths: torch.Tensor, g: Optional[torch.Tensor] = None
) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor]:
super().__call__(x, x_lengths, g=g)
return super().__call__(x, x_lengths, g=g)
def forward(
self, x: torch.Tensor, x_lengths: torch.Tensor, g: Optional[torch.Tensor] = None

4
rvc/layers/utils.py
View File

@@ -20,9 +20,9 @@ def slice_on_last_dim(
start_indices: List[int],
segment_size=4,
) -> torch.Tensor:
new_shape = x.shape
new_shape = [*x.shape]
new_shape[-1] = segment_size
ret = torch.empty(new_shape)
ret = torch.empty(new_shape, device=x.device)
for i in range(x.size(0)):
idx_str = start_indices[i]
idx_end = idx_str + segment_size