Update bert_gen.py

qymh233 · Sep 30, 2023 · 81ce0cc · 81ce0cc
1 parent 9587a63
commit 81ce0cc
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/bert_gen.py b/bert_gen.py
@@ -31,13 +31,13 @@ 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:
-        bert = get_bert(text, word2ph, language_str, device)
-        assert bert.shape[-1] == len(phone)
-        torch.save(bert, bert_path)
+    #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)
 
 
 if __name__ == "__main__":