From 01839af926ea1bd528e4e6489ca68105a5e8bfb8 Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Tue, 24 Oct 2023 18:30:14 -0300 Subject: [PATCH] Bug fix on XTTS masking training --- TTS/tts/layers/xtts/gpt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/tts/layers/xtts/gpt.py b/TTS/tts/layers/xtts/gpt.py index e7c0a41a..dfd7774e 100644 --- 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