Update data_utils.py

fishaudio · Sep 30, 2023 · 7b15665 · 7b15665
1 parent 1928118
commit 7b15665
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/data_utils.py b/data_utils.py
@@ -154,6 +154,7 @@ def get_text(self, text, word2ph, phone, tone, language_str, wav_path):
                 bert = torch.load(bert_path)
                 assert bert.shape[-1] == len(phone)
             elif language_str == "JP":
+                bert = torch.load(bert_path)
                 bert = get_bert_train(text_normalize(text), bert, word2ph, tokenizer)
                 assert bert.shape[-1] == len(phone), f"length of phonemes does not match input length of bert:{phone}, {bert.shape}"
         except Exception as e: