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( # bert, ja_bert, en_bert, phones, tones, lang_ids = get_text(
# text, language, hps, device # 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_start = (idx != 0) or (skip_start and idx == 0)
skip_end = (idx != len(text) - 1) or (skip_end and idx == len(text) - 1) skip_end = (idx != len(text) - 1) or (skip_end and idx == len(text) - 1)
( (
@@ -276,7 +276,7 @@ def infer_multilang(
temp_phones, temp_phones,
temp_tones, temp_tones,
temp_lang_ids, temp_lang_ids,
) = get_text(txt, ref, emot, lang, hps, device) ) = get_text(txt, ref, emotion, language, hps, device)
if skip_start: if skip_start:
temp_bert = temp_bert[:, 1:] temp_bert = temp_bert[:, 1:]
temp_ja_bert = temp_ja_bert[:, 1:] temp_ja_bert = temp_ja_bert[:, 1:]