tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Stardust·减
2023-11-07 14:30:56 +08:00
committed by GitHub
parent 2c2b7e85e1
commit d99b4bd721
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
data_utils.py
View File

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