更新 data_utils.py

This commit is contained in:
Stardust·减
2023-09-30 22:48:56 +08:00
committed by GitHub
parent ca04aad8c1
commit 0ea7530642

View File

@@ -155,7 +155,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
if language_str == "ZH": if language_str == "ZH":
bert = bert bert = bert
ja_bert = torch.zeros(768, len(phone)) ja_bert = torch.zeros(768, len(phone))
elif language_str == "JA": elif language_str == "JP":
ja_bert = bert ja_bert = bert
bert = torch.zeros(1024, len(phone)) bert = torch.zeros(1024, len(phone))
else: else: