tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Refactor: moved monotonic_align/ to style_bert_vits2/models/monotonic_alignment.py

Browse Source
This commit is contained in:
tsukumi
2024-03-07 04:36:05 +00:00
parent 1450bfd06f
commit d36401849b
7 changed files with 97 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
style_bert_vits2/models/models.py
View File

@@ -6,10 +6,10 @@ from torch.nn import Conv1d, Conv2d, ConvTranspose1d
from torch.nn import functional as F
from torch.nn.utils import remove_weight_norm, spectral_norm, weight_norm
import monotonic_align
from style_bert_vits2.models import attentions
from style_bert_vits2.models import commons
from style_bert_vits2.models import modules
from style_bert_vits2.models import monotonic_alignment
from style_bert_vits2.models.commons import get_padding, init_weights
from style_bert_vits2.text_processing.symbols import NUM_LANGUAGES, NUM_TONES, SYMBOLS
@@ -932,7 +932,7 @@ class SynthesizerTrn(nn.Module):
attn_mask = torch.unsqueeze(x_mask, 2) * torch.unsqueeze(y_mask, -1)
attn = (
monotonic_align.maximum_path(neg_cent, attn_mask.squeeze(1))
monotonic_alignment.maximum_path(neg_cent, attn_mask.squeeze(1))
.unsqueeze(1)
.detach()
)