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

Dockerfile

@@ -1,5 +1,5 @@
-# Use ROCm PyTorch base image
-FROM rocm/pytorch:latest
+# Use ROCm PyTorch base image with compatible PyTorch
+FROM rocm/pytorch:2.4.0-rocm6.0-cxx11-ubuntu22.04
 
 # Set environment variables for ROCm and Python
 ENV PYTHONDONTWRITEBYTECODE=1

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",