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

Update infer.py

Browse Source
This commit is contained in:
Stardust·减
2023-11-07 17:56:28 +08:00
committed by GitHub
parent 17ae5b9cf5
commit 03b2ec09e1
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
infer.py
View File

@@ -85,22 +85,22 @@ def get_text(text, language_str, hps, device):
for i in range(len(word2ph)): for i in range(len(word2ph)):
word2ph[i] = word2ph[i] * 2 word2ph[i] = word2ph[i] * 2
word2ph[0] += 1 word2ph[0] += 1
bert = get_bert(norm_text, word2ph, language_str, device) bert_ori = get_bert(norm_text, word2ph, language_str, device)
del word2ph del word2ph
assert bert.shape[-1] == len(phone), phone assert bert_ori.shape[-1] == len(phone), phone
if language_str == "ZH": if language_str == "ZH":
bert = bert bert = bert_ori
ja_bert = torch.zeros(1024, len(phone)) ja_bert = torch.zeros(1024, len(phone))
en_bert = torch.zeros(1024, len(phone)) en_bert = torch.zeros(1024, len(phone))
elif language_str == "JP": elif language_str == "JP":
bert = torch.zeros(1024, len(phone)) bert = torch.zeros(1024, len(phone))
ja_bert = bert ja_bert = bert_ori
en_bert = torch.zeros(1024, len(phone)) en_bert = torch.zeros(1024, len(phone))
elif language_str == "EN": elif language_str == "EN":
bert = torch.zeros(1024, len(phone)) bert = torch.zeros(1024, len(phone))
ja_bert = torch.zeros(1024, len(phone)) ja_bert = torch.zeros(1024, len(phone))
en_bert = bert en_bert = bert_ori
else: else:
raise ValueError("language_str should be ZH, JP or EN") raise ValueError("language_str should be ZH, JP or EN")