Update data_utils.py

fishaudio · Sep 30, 2023 · 577a7de · 577a7de
1 parent c87eebb
commit 577a7de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data_utils.py b/data_utils.py
@@ -147,7 +147,7 @@ def get_text(self, text, word2ph, phone, tone, language_str, wav_path):
         try:
             bert = torch.load(bert_path)
         except:
-            bert = get_bert(text, word2ph, language_str,device=None)
+            bert = get_bert(text, word2ph, language_str, "cuda", tokenizer)
             torch.save(bert, bert_path)
 
         if language_str == "ZH":