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

修复bug (#2704)

Browse Source 
* 修复bug

* fallbak and bug fix
This commit is contained in:
ChasonJiang 2025-12-30 16:00:21 +08:00 committed by GitHub
parent 9080a967d5
commit c767f0b83b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 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

@ -707,12 +707,11 @@ class Text2SemanticDecoder(nn.Module):
if idx == 0: if idx == 0:
attn_mask = F.pad(attn_mask[:, :, -1].unsqueeze(-2), (0, 1), value=False) attn_mask = F.pad(attn_mask[:, :, -1].unsqueeze(-2), (0, 1), value=False)
logits = logits[:, :-1]
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） if idx < 11: ###至少预测出10个token不然不给停止（0.4s）
logits[:, -1] = float("-inf") 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