Update bert_gen.py

2023-09-30 19:51:02 +08:00
parent 81ce0cc1f7
commit 89f2f4cdd5
1 changed files with 7 additions and 7 deletions
--- a/bert_gen.py
+++ b/bert_gen.py
@@ -31,10 +31,10 @@ def process_line(line):

    bert_path = wav_path.replace(".wav", ".bert.pt")

-    #try:
-        #bert = torch.load(bert_path)
-        #assert bert.shape[-1] == len(phone)
-    #except Exception:
+    try:
+        bert = torch.load(bert_path)
+        assert bert.shape[-1] == len(phone)
+    except Exception:
        bert = get_bert(text, word2ph, language_str, device)
        assert bert.shape[-1] == len(phone)
        torch.save(bert, bert_path)