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

Add files via upload

Browse Source
This commit is contained in:
Ναρουσέ·μ·γιουμεμί·Χινακάννα 2024-02-08 21:41:37 +08:00 committed by GitHub
parent 6dde4ac622
commit 84622abd24
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -57,7 +57,7 @@ def logits_to_probs(
logits = logits / max(temperature, 1e-5) logits = logits / max(temperature, 1e-5)
if top_k is not None: if top_k is not None:
v, _ = torch.topk(logits, min(top_k, logits.size(-1))) v, _ = torch.topk(logits, top_k)
pivot = v.select(-1, -1).unsqueeze(-1) pivot = v.select(-1, -1).unsqueeze(-1)
logits = torch.where(logits < pivot, inf_tensor_value, logits) logits = torch.where(logits < pivot, inf_tensor_value, logits)