fix: idx error when do repetition penalty (#738)

Co-authored-by: Ubuntu <[email protected]>
2noise · Aug 31, 2024 · 3990f0e · 3990f0e
1 parent a82e1df
commit 3990f0e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ChatTTS/model/velocity/model_runner.py b/ChatTTS/model/velocity/model_runner.py
@@ -526,6 +526,7 @@ def execute_model(
                     for i in range(self.post_model.num_vq)
                 ]
                 input_emb = torch.stack(code_emb, 3).sum(3)
+                start_idx = input_tokens_history.shape[-2] - 1 if input_tokens_history.shape[-2] > 0 else 0 
         else:
             input_emb = self.post_model(input_tokens, text_mask)
         # print(input_emb.shape)