tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update infer.py

Browse Source
This commit is contained in:
Stardust·减
2023-11-26 09:30:51 +08:00
committed by GitHub
parent 2e73312808
commit 4158626b50
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
infer.py
View File

@@ -265,7 +265,7 @@ def infer_multilang(
# bert, ja_bert, en_bert, phones, tones, lang_ids = get_text(
# text, language, hps, device
# )
for idx, (txt, lang, ref, emot) in enumerate(zip(text, language, reference_audio, emotion)):
for idx, (txt, lang) in enumerate(zip(text, language)):
skip_start = (idx != 0) or (skip_start and idx == 0)
skip_end = (idx != len(text) - 1) or (skip_end and idx == len(text) - 1)
(
@@ -276,7 +276,7 @@ def infer_multilang(
temp_phones,
temp_tones,
temp_lang_ids,
) = get_text(txt, ref, emot, lang, hps, device)
) = get_text(txt, ref, emotion, language, hps, device)
if skip_start:
temp_bert = temp_bert[:, 1:]
temp_ja_bert = temp_ja_bert[:, 1:]