From 98126b4668278707c4518d6dc2bf16d0654beb2d Mon Sep 17 00:00:00 2001 From: ChasonJiang <1440499136@qq.com> Date: Fri, 19 Dec 2025 17:17: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?= 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]