Update bert_gen.py

This commit is contained in:
Stardust·减
2023-09-30 19:49:44 +08:00
committed by GitHub
parent 9587a6361b
commit 81ce0cc1f7

View File

@@ -31,13 +31,13 @@ def process_line(line):
bert_path = wav_path.replace(".wav", ".bert.pt") bert_path = wav_path.replace(".wav", ".bert.pt")
try: #try:
bert = torch.load(bert_path) #bert = torch.load(bert_path)
assert bert.shape[-1] == len(phone) #assert bert.shape[-1] == len(phone)
except Exception: #except Exception:
bert = get_bert(text, word2ph, language_str, device) bert = get_bert(text, word2ph, language_str, device)
assert bert.shape[-1] == len(phone) assert bert.shape[-1] == len(phone)
torch.save(bert, bert_path) torch.save(bert, bert_path)
if __name__ == "__main__": if __name__ == "__main__":