fix

2023-09-09 11:36:50 +08:00
parent 7a25c8afb0
commit f69599cb69
1 changed files with 1 additions and 1 deletions
--- a/webui.py
+++ b/webui.py
@@ -43,7 +43,7 @@ def get_text(text, language_str, hps):
        for i in range(len(word2ph)):
            word2ph[i] = word2ph[i] * 2
        word2ph[0] += 1
-    bert = get_bert(norm_text, word2ph, language_str)
+    bert = get_bert(norm_text, word2ph, language_str, device)
    del word2ph
    assert bert.shape[-1] == len(phone), phone