Update t2s_model.py

2024-09-10 19:13:42 +08:00
parent 3488cffd68
commit 40cd22e69d
1 changed files with 1 additions and 0 deletions
--- a/GPT_SoVITS/AR/models/t2s_model.py
+++ b/GPT_SoVITS/AR/models/t2s_model.py
@@ -854,6 +854,7 @@ class Text2SemanticDecoder(nn.Module):

            if idx == 0:
                xy_attn_mask = None
+            if(idx<11):###至少预测出10个token不然不给停止（0.4s）
                logits = logits[:, :-1]

            samples = sample(