This commit is contained in:
Stardust·减
2023-09-09 11:36:50 +08:00
committed by GitHub
parent 7a25c8afb0
commit f69599cb69

View File

@@ -43,7 +43,7 @@ def get_text(text, language_str, hps):
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) bert = get_bert(norm_text, word2ph, language_str, device)
del word2ph del word2ph
assert bert.shape[-1] == len(phone), phone assert bert.shape[-1] == len(phone), phone