Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Bug fix on XTTS masking training

This commit is contained in:
Edresson Casanova 2023-10-24 18:30:14 -03:00
parent 8af3d2dbcd
commit 01839af926
1 changed files with 1 additions and 1 deletions
TTS/tts/layers/xtts
gpt.py

2
TTS/tts/layers/xtts/gpt.py
View File

@ -450,7 +450,7 @@ class GPT(nn.Module):
) )
if cond_idxs is not None: if cond_idxs is not None:
for idx, r in enumerate(cond_idxs.squeeze()): for idx, r in enumerate(cond_idxs):
l = r[1] - r[0] l = r[1] - r[0]
attn_mask_cond[idx, l:] = 0.0 attn_mask_cond[idx, l:] = 0.0