Update data_utils.py

This commit is contained in:
Stardust·减
2023-09-06 10:12:31 +08:00
committed by GitHub
parent 0afb102d48
commit 57276857d9

View File

@@ -80,7 +80,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
spec, wav = self.get_audio(audiopath) spec, wav = self.get_audio(audiopath)
sid = torch.LongTensor([int(self.spk_map[sid])]) sid = torch.LongTensor([int(self.spk_map[sid])])
return (phones, spec, wav, sid, tone, language, bertja_bert) return (phones, spec, wav, sid, tone, language, bert, ja_bert)
def get_audio(self, filename): def get_audio(self, filename):
audio, sampling_rate = load_wav_to_torch(filename) audio, sampling_rate = load_wav_to_torch(filename)