Update data_utils.py

This commit is contained in:
Stardust·减
2023-11-09 17:30:05 +08:00
committed by GitHub
parent f15f083523
commit 653a1861cb

View File

@@ -148,8 +148,8 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
bert_ori = torch.load(bert_path) bert_ori = torch.load(bert_path)
assert bert_ori.shape[-1] == len(phone) assert bert_ori.shape[-1] == len(phone)
except Exception as e: except Exception as e:
logger.warn("Bert load Failed") logger.warning("Bert load Failed")
logger.warn(e) logger.warning(e)
if language_str == "ZH": if language_str == "ZH":
bert = bert_ori bert = bert_ori