diff --git a/pyproject.toml b/pyproject.toml index d010b92..9d092d3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -58,9 +58,9 @@ gpu-deps = [ # 34f52344e5ba96660fac877393e1954e399d3ef3 (matches # facebookresearch/tribev2 main pyproject.toml). TRIBE itself is # installed separately from requirements/tribev2-gpu.txt. - "torch>=2.5.1,<2.7", - "torchvision>=0.20,<0.22", - "torchaudio>=2.5.1,<2.7", # TRIBE's audio (W2V-BERT) forward pass needs it + "torch>=2.5.1,<2.13", + "torchvision>=0.20,<0.28", + "torchaudio>=2.5.1,<2.12", # TRIBE's audio (W2V-BERT) forward pass needs it "transformers>=4.45", "huggingface-hub>=0.24", "moviepy>=2.2.1", diff --git a/requirements/tribev2-gpu.txt b/requirements/tribev2-gpu.txt index eb29739..1817e20 100644 --- a/requirements/tribev2-gpu.txt +++ b/requirements/tribev2-gpu.txt @@ -17,4 +17,4 @@ tribev2 @ git+https://github.com/facebookresearch/tribev2.git@34f52344e5ba96660f # `uvx whisperx`, but pinning it here keeps the full `score()` path # reproducible (and makes a non-uvx environment work too). torch / # torchaudio / transformers etc. are declared in the `[gpu-deps]` extra. -whisperx>=3.1.1 +whisperx>=3.8.6