Update t2s_model.py

This commit is contained in:
RVC-Boss 2024-09-10 19:13:42 +08:00 committed by GitHub
parent 3488cffd68
commit 40cd22e69d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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(