From d99b4bd72169fd266c023299b3d4a5214c2a4ea1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Tue, 7 Nov 2023 14:30:56 +0800
Subject: [PATCH] fix

---
 data_utils.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/data_utils.py b/data_utils.py
index d6526c4..9589d56 100644
--- a/data_utils.py
+++ b/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),