Bug fix on XTTS masking training

2023-10-24 18:30:14 -03:00 · 2023-10-24 18:30:14 -03:00 · 01839af926
parent 8af3d2dbcd
commit 01839af926
1 changed files with 1 additions and 1 deletions
--- a/TTS/tts/layers/xtts/gpt.py
+++ b/TTS/tts/layers/xtts/gpt.py
@ -450,7 +450,7 @@ class GPT(nn.Module):
            )

            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]
                    attn_mask_cond[idx, l:] = 0.0