Update data_utils.py

fishaudio · Sep 30, 2023 · 9587a63 · 9587a63
1 parent fb54a24
commit 9587a63
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/data_utils.py b/data_utils.py
@@ -7,7 +7,7 @@
 import commons
 from mel_processing import spectrogram_torch, mel_spectrogram_torch
 from utils import load_wav_to_torch, load_filepaths_and_text
-from text import cleaned_text_to_sequence
+from text import cleaned_text_to_sequence，get_bert
 
 """Multi speaker version"""
 
@@ -148,8 +148,8 @@ def get_text(self, text, word2ph, phone, tone, language_str, wav_path):
             bert = torch.load(bert_path)
             assert bert.shape[-1] == len(phone)
         except:
-            #bert = get_bert(text, word2ph, language_str)
-            #torch.save(bert, bert_path)
+            bert = get_bert(text, word2ph, language_str)
+            torch.save(bert, bert_path)
             assert bert.shape[-1] == len(phone), phone
 
         if language_str == "ZH":