From 4158626b50d46b20cd79d9cd6783a4a08a460817 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Sun, 26 Nov 2023 09:30:51 +0800 Subject: [PATCH] Update infer.py --- infer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/infer.py b/infer.py index 598deb9..b856091 100644 --- a/infer.py +++ b/infer.py @@ -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:]