Update bert_gen.py

fishaudio · Sep 30, 2023 · 9169311 · 9169311
1 parent 4e4a4d2
commit 9169311
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bert_gen.py b/bert_gen.py
@@ -36,7 +36,7 @@ def process_line(line):
         assert bert.shape[-1] == len(phone)
     except Exception:
         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)