blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-12 19:39:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

改名 export_torch_script_v3.py 为 export_torch_script_v3v4.py

Browse Source
This commit is contained in:
csh 2025-05-29 02:42:22 +08:00
parent 150cd842df
commit d2b5db9f2c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
GPT_SoVITS/export_torch_script_v3.py → GPT_SoVITS/export_torch_script_v3v4.py
View File

@ -473,7 +473,7 @@ class GPTSoVITSV4(torch.nn.Module):
# 因为导出的模型在不同shape时会重新编译还是怎么的，会卡顿10s这样，
# 所以在这里补0让他shape维持不变
# 但是这样会导致生成的音频长度不对，所以在最后截取一下。
# 经过 bigvgan 之后音频长度就是 fea_todo.shape[2] * 256
# 经过 hifigan 之后音频长度就是 fea_todo.shape[2] * 480
complete_len = chunk_len - fea_todo_chunk.shape[-1]
if complete_len != 0:
fea_todo_chunk = torch.cat(
@ -1090,7 +1090,7 @@ def test_export1(
bert1 = bert1.T.to(device)
bert2 = bert2.T.to(device)
top_k = torch.LongTensor([15]).to(device)
top_k = torch.LongTensor([20]).to(device)
current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
logger.info("start inference %s", current_time)