m-bain/whisperX

Potential optimisation: temp-file mode for load_audio() to reduce memory on long audio

开放

#1,440 创建于 2026年6月20日

 (2 条评论) (0 个反应) (0 位负责人)Python (2,386 个派生)batch import
enhancementhelp wanted

仓库指标

星标
 (23,718 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

load_audio() currently buffers the entire ffmpeg output in Python memory via subprocess.run(capture_output=True). For very long audio (10h+ ≈ 1.15 GB raw PCM), this can cause OOM on constrained systems since the Python bytes buffer, numpy view, and float32 copy all coexist at peak.

A use_tmp_file=True option could write ffmpeg output to a temp file instead of piping to stdout, then read with np.fromfile(), keeping the raw PCM out of Python's heap entirely.

Originally proposed in #1221 (bundled with unrelated dep changes that have since landed).

贡献者指南