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

Update data_utils.py

Browse Source
This commit is contained in:
Stardust·减
2023-09-30 12:53:44 +08:00
committed by GitHub
parent 61130ac803
commit fb54a24ae3
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
data_utils.py
View File

@@ -7,7 +7,7 @@ from loguru import logger
import commons import commons
from mel_processing import spectrogram_torch, mel_spectrogram_torch from mel_processing import spectrogram_torch, mel_spectrogram_torch
from utils import load_wav_to_torch, load_filepaths_and_text 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""" """Multi speaker version"""
@@ -148,8 +148,8 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
bert = torch.load(bert_path) bert = torch.load(bert_path)
assert bert.shape[-1] == len(phone) assert bert.shape[-1] == len(phone)
except: except:
bert = get_bert(text, word2ph, language_str) #bert = get_bert(text, word2ph, language_str)
torch.save(bert, bert_path) #torch.save(bert, bert_path)
assert bert.shape[-1] == len(phone), phone assert bert.shape[-1] == len(phone), phone
if language_str == "ZH": if language_str == "ZH":