Skip to content

Commit 2c528ce

Browse files
authored
Merge pull request #198 from OedoSoldier/master
Update infer.py
2 parents 818b3b1 + 8ef5db5 commit 2c528ce

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

infer.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def get_text(text, language_str, hps, device):
123123
return bert, ja_bert, en_bert, phone, tone, language
124124

125125

126-
def get_emo(reference_audio, emotion):
126+
def get_emo_(reference_audio, emotion):
127127
emo = (
128128
torch.from_numpy(get_emo(reference_audio))
129129
if reference_audio
@@ -197,7 +197,7 @@ def infer(
197197
bert, ja_bert, en_bert, phones, tones, lang_ids = get_text(
198198
text, language, hps, device
199199
)
200-
emo = get_emo(reference_audio, emotion)
200+
emo = get_emo_(reference_audio, emotion)
201201
if skip_start:
202202
phones = phones[1:]
203203
tones = tones[1:]
@@ -266,7 +266,7 @@ def infer_multilang(
266266
skip_end=False,
267267
):
268268
bert, ja_bert, en_bert, phones, tones, lang_ids = [], [], [], [], [], []
269-
emo = get_emo(reference_audio, emotion)
269+
emo = get_emo_(reference_audio, emotion)
270270
for idx, (txt, lang) in enumerate(zip(text, language)):
271271
skip_start = (idx != 0) or (skip_start and idx == 0)
272272
skip_end = (idx != len(text) - 1) or (skip_end and idx == len(text) - 1)

0 commit comments

Comments
 (0)