Update bert_gen.py

fishaudio · Jan 4, 2024 · baa61d2 · baa61d2
1 parent 62d3a2e
commit baa61d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bert_gen.py b/bert_gen.py
@@ -39,7 +39,7 @@ def process_line(x):
 
     try:
         bert = torch.load(bert_path)
-        assert bert.shape[-1] == len(phone)
+        assert bert.shape[0] == 2048
     except Exception:
         bert = get_bert(text, word2ph, language_str, device)
         assert bert.shape[-1] == len(phone)