fix

2026-07-23 11:51:07 +09:00
parent 4f3bcfd252
commit 9cfaa27f3f
5 changed files with 87 additions and 17 deletions
--- a/bert_gen.py
+++ b/bert_gen.py
@@ -55,7 +55,7 @@ def process_line(x: tuple[str, bool]):
    bert_path = wav_path.replace(".WAV", ".wav").replace(".wav", ".bert.pt")

    try:
-        bert = torch.load(bert_path)
+        bert = torch.load(bert_path, weights_only=True)
        assert bert.shape[-1] == len(phone)
    except Exception:
        bert = extract_bert_feature(text, word2ph, Languages(language_str), device)