Fix PyTorch version compatibility and update ROCm Dockerfile image

- Change torch to stable 2.4.0, add torchvision for compatibility - Update Dockerfile to use specific ROCm PyTorch image version
2026-05-13 02:15:48 +03:00
parent 67da25d849
commit 221ecbe69d
2 changed files with 5 additions and 4 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,8 +19,9 @@ dependencies = [
    "av<16.0.0",
    "numpy>=2.1.0,<2.3.0; python_version >='3.13'",
    "pyannote-audio>=3.3.2,<4.0.0",
-    "torch~=2.8.0",
-    "torchaudio~=2.8.0",
+    "torch~=2.4.0",
+    "torchaudio~=2.4.0",
+    "torchvision~=0.19.0",
    "transformers>=4.48.0",
    "triton>=3.3.0; sys_platform == 'linux' and platform_machine == 'x86_64'", # only install triton on x86_64 Linux
    "fastapi>=0.104.0",