Add train_narrator_tts.py

train_narrator_tts.py

@@ -214,6 +214,17 @@ def push_model(ckpt_dir: Path) -> None:
 
 
 if __name__ == "__main__":
+    # torchcodec (datasets audio decoder) needs FFmpeg system libs — install before training
+    print("Installing FFmpeg for torchcodec audio decoding...")
+    r = subprocess.run(
+        ["apt-get", "install", "-y", "--no-install-recommends", "ffmpeg"],
+        capture_output=True, text=True,
+    )
+    if r.returncode == 0:
+        print("FFmpeg installed OK")
+    else:
+        print(f"apt-get failed (rc={r.returncode}): {r.stderr[-300:]}")
+
     prepare_data()
     ckpt_dir = fine_tune()
     push_model(ckpt_dir)