diff --git a/requirements.txt b/requirements.txt index 93e2b8e4..6117e5fa 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,19 @@ +# Requires Python <=3.11 (numpy 1.24.4 does not build on Python 3.12+) + +# Pin torch/torchaudio — must match a version that torbi ships prebuilt binaries for +# (2.6, 2.7, or 2.8) and that pyannote.audio 3.3.0 supports +torch==2.6.0 +torchaudio==2.6.0 + +# Pin numpy — required by brouhaha-vad (pinned to ==1.24.4) +numpy==1.24.4 + +# Pin scipy/scikit-learn/torchmetrics to versions compatible with numpy 1.24 +scipy<1.12 +scikit-learn<1.4 +torchmetrics<1.3 + +# Direct dependencies datasets[audio] https://github.com/marianne-m/brouhaha-vad/archive/main.zip penn