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

Merge pull request #198 from OedoSoldier/master

Browse Source 
Update infer.py
This commit is contained in:
OedoSoldier
2023-11-27 13:14:32 +08:00
committed by GitHub
parent 818b3b11c3 8ef5db527b
commit 2c528ce130
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
infer.py
View File

@@ -123,7 +123,7 @@ def get_text(text, language_str, hps, device):
return bert, ja_bert, en_bert, phone, tone, language return bert, ja_bert, en_bert, phone, tone, language
def get_emo(reference_audio, emotion): def get_emo_(reference_audio, emotion):
emo = ( emo = (
torch.from_numpy(get_emo(reference_audio)) torch.from_numpy(get_emo(reference_audio))
if reference_audio if reference_audio
@@ -197,7 +197,7 @@ def infer(
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
) )
emo = get_emo(reference_audio, emotion) emo = get_emo_(reference_audio, emotion)
if skip_start: if skip_start:
phones = phones[1:] phones = phones[1:]
tones = tones[1:] tones = tones[1:]
@@ -266,7 +266,7 @@ def infer_multilang(
skip_end=False, skip_end=False,
): ):
bert, ja_bert, en_bert, phones, tones, lang_ids = [], [], [], [], [], [] bert, ja_bert, en_bert, phones, tones, lang_ids = [], [], [], [], [], []
emo = get_emo(reference_audio, emotion) emo = get_emo_(reference_audio, emotion)
for idx, (txt, lang) in enumerate(zip(text, language)): 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)