From f69599cb69e8564d72d68f5922208a74d8051601 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Sat, 9 Sep 2023 11:36:50 +0800 Subject: [PATCH] fix --- webui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/webui.py b/webui.py index 7464eb2..20f472b 100644 --- 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