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),