From 6f3d1ec7e5e62cde88bf9250afc4e014cacafa70 Mon Sep 17 00:00:00 2001 From: wassname Date: Sat, 7 Oct 2023 13:51:47 +0800 Subject: [PATCH] possible fix for cuda assert error --- TTS/tts/layers/xtts/gpt_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/tts/layers/xtts/gpt_inference.py b/TTS/tts/layers/xtts/gpt_inference.py index d44bd3de..b2e6c138 100644 --- a/TTS/tts/layers/xtts/gpt_inference.py +++ b/TTS/tts/layers/xtts/gpt_inference.py @@ -92,7 +92,7 @@ class GPT2InferenceModel(GPT2PreTrainedModel): else: emb = self.embeddings(input_ids) emb = emb + self.pos_embedding.get_fixed_embedding( - attention_mask.shape[1] - (prefix_len + 1), attention_mask.device + attention_mask.shape[1] - (prefix_len + 2), attention_mask.device ) transformer_outputs = self.transformer( inputs_embeds=emb,