storytold/vits-finetuning
2
0
Fork 0
mirror of https://github.com/storytold/vits-finetuning.git synced 2026-10-17 00:23:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Bin loss, which increases naturalness

Browse Source
This commit is contained in:
ZDisket
2023-06-25 19:18:56 -03:00
parent 87d5138a3c
commit b1e03bee38
6 changed files with 48 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models.py
+1 -1
View File
@@ -734,7 +734,7 @@ class SynthesizerTrn(nn.Module):
spec, phase, x1 = self.dec(z_slice, g=g)
o = self.stft.inverse(spec, phase)
return o, x1, l_length, attn, attn_logprob, ids_slice, x_mask, y_mask, (z, z_p, m_p, logs_p, m_q, logs_q)
return o, x1, l_length, attn, attn_logprob, attn_soft, ids_slice, x_mask, y_mask, (z, z_p, m_p, logs_p, m_q, logs_q)
def infer(self, x, x_lengths, tm_hidden, bert, bert_lens, sid=None, noise_scale=1, length_scale=1, noise_scale_w=1., max_len=None):
tm_encoded = self.tm_enc(tm_hidden) # [b, tm_last_size]