blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2026-01-01 13:47:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

对齐naive_infer的解码策略，防止吞句 (#2697)

Browse Source
This commit is contained in:
ChasonJiang 2025-12-19 17:37:19 +08:00 committed by GitHub
parent abe984395c
commit bfca0f6b2d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
GPT_SoVITS/AR/models/t2s_model.py
View File

@ -711,6 +711,9 @@ class Text2SemanticDecoder(nn.Module):
else: else:
attn_mask = F.pad(attn_mask, (0, 1), value=False) attn_mask = F.pad(attn_mask, (0, 1), value=False)
if idx < 11: ###至少预测出10个token不然不给停止（0.4s）
logits = logits[:, :-1]
samples = sample( samples = sample(
logits, y, top_k=top_k, top_p=top_p, repetition_penalty=repetition_penalty, temperature=temperature logits, y, top_k=top_k, top_p=top_p, repetition_penalty=repetition_penalty, temperature=temperature
)[0] )[0]