blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-26 03:26:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update t2s_model.py

Browse Source
This commit is contained in:
RVC-Boss 2024-01-22 23:26:18 +08:00 committed by GitHub
parent 1dba692448
commit 872134c846
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -302,6 +302,8 @@ class Text2SemanticDecoder(nn.Module):
xy_dec[:, -1] xy_dec[:, -1]
) ##不用改，如果用了cache的默认就是只有一帧，取最后一帧一样的 ) ##不用改，如果用了cache的默认就是只有一帧，取最后一帧一样的
# samples = topk_sampling(logits, top_k=top_k, top_p=1.0, temperature=temperature) # samples = topk_sampling(logits, top_k=top_k, top_p=1.0, temperature=temperature)
if(idx==0):###第一次跑不能EOS否则没有了
logits = logits[:, :-1] ###刨除1024终止符号的概率
samples = sample( samples = sample(
logits[0], y, top_k=top_k, top_p=1.0, repetition_penalty=1.35 logits[0], y, top_k=top_k, top_p=1.0, repetition_penalty=1.35
)[0].unsqueeze(0) )[0].unsqueeze(0)