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

Fix multilang generation

Browse Source
This commit is contained in:
Sihan Wang
2023-12-09 23:25:59 +08:00
parent f1a1b4c73d
commit 6733e63b31
1 changed files with 24 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
infer.py
View File

@@ -199,19 +199,19 @@ def infer(
) )
emo = get_emo_(reference_audio, emotion) emo = get_emo_(reference_audio, emotion)
if skip_start: if skip_start:
phones = phones[1:] phones = phones[3:]
tones = tones[1:] tones = tones[3:]
lang_ids = lang_ids[1:] lang_ids = lang_ids[3:]
bert = bert[:, 1:] bert = bert[:, 3:]
ja_bert = ja_bert[:, 1:] ja_bert = ja_bert[:, 3:]
en_bert = en_bert[:, 1:] en_bert = en_bert[:, 3:]
if skip_end: if skip_end:
phones = phones[:-1] phones = phones[:-2]
tones = tones[:-1] tones = tones[:-2]
lang_ids = lang_ids[:-1] lang_ids = lang_ids[:-2]
bert = bert[:, :-1] bert = bert[:, :-2]
ja_bert = ja_bert[:, :-1] ja_bert = ja_bert[:, :-2]
en_bert = en_bert[:, :-1] en_bert = en_bert[:, :-2]
with torch.no_grad(): with torch.no_grad():
x_tst = phones.to(device).unsqueeze(0) x_tst = phones.to(device).unsqueeze(0)
tones = tones.to(device).unsqueeze(0) tones = tones.to(device).unsqueeze(0)
@@ -279,19 +279,19 @@ def infer_multilang(
temp_lang_ids, temp_lang_ids,
) = get_text(txt, lang, hps, device) ) = get_text(txt, lang, hps, device)
if skip_start: if skip_start:
temp_bert = temp_bert[:, 1:] temp_bert = temp_bert[:, 3:]
temp_ja_bert = temp_ja_bert[:, 1:] temp_ja_bert = temp_ja_bert[:, 3:]
temp_en_bert = temp_en_bert[:, 1:] temp_en_bert = temp_en_bert[:, 3:]
temp_phones = temp_phones[1:] temp_phones = temp_phones[3:]
temp_tones = temp_tones[1:] temp_tones = temp_tones[3:]
temp_lang_ids = temp_lang_ids[1:] temp_lang_ids = temp_lang_ids[3:]
if skip_end: if skip_end:
temp_bert = temp_bert[:, :-1] temp_bert = temp_bert[:, :-2]
temp_ja_bert = temp_ja_bert[:, :-1] temp_ja_bert = temp_ja_bert[:, :-2]
temp_en_bert = temp_en_bert[:, :-1] temp_en_bert = temp_en_bert[:, :-2]
temp_phones = temp_phones[:-1] temp_phones = temp_phones[:-2]
temp_tones = temp_tones[:-1] temp_tones = temp_tones[:-2]
temp_lang_ids = temp_lang_ids[:-1] temp_lang_ids = temp_lang_ids[:-2]
bert.append(temp_bert) bert.append(temp_bert)
ja_bert.append(temp_ja_bert) ja_bert.append(temp_ja_bert)
en_bert.append(temp_en_bert) en_bert.append(temp_en_bert)