fix

data_utils.py

@@ -145,24 +145,24 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
             word2ph[0] += 1
         bert_path = wav_path.replace(".wav", ".bert.pt")
         try:
-            bert = torch.load(bert_path)
-            assert bert.shape[-1] == len(phone)
+            bert_ori = torch.load(bert_path)
+            assert bert_ori.shape[-1] == len(phone)
         except Exception as e:
             logger.warn("Bert load Failed")
             logger.warn(e)
 
         if language_str == "ZH":
-            bert = bert
+            bert = bert_ori
             ja_bert = torch.zeros(1024, len(phone))
             en_bert = torch.zeros(1024, len(phone))
         elif language_str == "JP":
             bert = torch.zeros(1024, len(phone))
-            ja_bert = bert
+            ja_bert = bert_ori
             en_bert = torch.zeros(1024, len(phone))
         elif language_str == "EN":
             bert = torch.zeros(1024, len(phone))
             ja_bert = torch.zeros(1024, len(phone))
-            en_bert = bert
+            en_bert = bert_ori
         assert bert.shape[-1] == len(phone), (
             bert.shape,
             len(phone),