Update infer.py

This commit is contained in:
Stardust·减
2023-11-27 12:49:16 +08:00
committed by GitHub
parent dec3fc0737
commit 1952bcd40e

View File

@@ -276,12 +276,11 @@ def infer_multilang(
temp_phones, temp_phones,
temp_tones, temp_tones,
temp_lang_ids, temp_lang_ids,
) = get_text(txt, reference_audio, emotion, language, hps, device) ) = get_text(txt, reference_audio, emotion, lang, 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:]
temp_en_bert = temp_en_bert[:, 1:] temp_en_bert = temp_en_bert[:, 1:]
temp_emo = temp_emo[:, 1:]
temp_phones = temp_phones[1:] temp_phones = temp_phones[1:]
temp_tones = temp_tones[1:] temp_tones = temp_tones[1:]
temp_lang_ids = temp_lang_ids[1:] temp_lang_ids = temp_lang_ids[1:]
@@ -289,7 +288,6 @@ def infer_multilang(
temp_bert = temp_bert[:, :-1] temp_bert = temp_bert[:, :-1]
temp_ja_bert = temp_ja_bert[:, :-1] temp_ja_bert = temp_ja_bert[:, :-1]
temp_en_bert = temp_en_bert[:, :-1] temp_en_bert = temp_en_bert[:, :-1]
temp_emo = temp_emo[:, :-1]
temp_phones = temp_phones[:-1] temp_phones = temp_phones[:-1]
temp_tones = temp_tones[:-1] temp_tones = temp_tones[:-1]
temp_lang_ids = temp_lang_ids[:-1] temp_lang_ids = temp_lang_ids[:-1]