From bfca0f6b2dd9f846c76366be807f01a8873140a0 Mon Sep 17 00:00:00 2001
From: ChasonJiang <46401978+ChasonJiang@users.noreply.github.com>
Date: Fri, 19 Dec 2025 17:37:19 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AF=B9=E9=BD=90naive=5Finfer=E7=9A=84?=
 =?UTF-8?q?=E8=A7=A3=E7=A0=81=E7=AD=96=E7=95=A5=EF=BC=8C=E9=98=B2=E6=AD=A2?=
 =?UTF-8?q?=E5=90=9E=E5=8F=A5=20(#2697)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 GPT_SoVITS/AR/models/t2s_model.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/GPT_SoVITS/AR/models/t2s_model.py b/GPT_SoVITS/AR/models/t2s_model.py
index 0caadd04..81aad1ec 100644
--- a/GPT_SoVITS/AR/models/t2s_model.py
+++ b/GPT_SoVITS/AR/models/t2s_model.py
@@ -711,6 +711,9 @@ class Text2SemanticDecoder(nn.Module):
             else:
                 attn_mask = F.pad(attn_mask, (0, 1), value=False)
 
+            if idx < 11:  ###至少预测出10个token不然不给停止（0.4s）
+                logits = logits[:, :-1]
+
             samples = sample(
                 logits, y, top_k=top_k, top_p=top_p, repetition_penalty=repetition_penalty, temperature=temperature
             )[0]