Update data_utils.py

This commit is contained in:
Stardust·减
2023-08-21 09:23:00 +08:00
committed by GitHub
parent ea4a48db11
commit fdf620b0c4

View File

@@ -8,7 +8,7 @@ import torch.utils.data
import commons
from mel_processing import spectrogram_torch, mel_spectrogram_torch, spec_to_mel_torch
from utils import load_wav_to_torch, load_filepaths_and_text
from text import cleaned_text_to_sequence, get_bert
from text import cleaned_text_to_sequence
"""Multi speaker version"""
@@ -135,10 +135,10 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
bert = torch.load(bert_path)
assert bert.shape[-1] == len(phone)
except:
bert = get_bert(text, word2ph, language_str)
torch.save(bert, bert_path)
#bert = get_bert(text, word2ph, language_str)
#torch.save(bert, bert_path)
print(bert.shape[-1], bert_path, text, pold)
assert bert.shape[-1] == len(phone)
#assert bert.shape[-1] == len(phone)
assert bert.shape[-1] == len(phone), (
bert.shape, len(phone), sum(word2ph), p1, p2, t1, t2, pold, pold2, word2ph, text, w2pho)