vlad.os/whisperx-rocm-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix PyTorch version compatibility and update ROCm Dockerfile image

Browse Source 
- Change torch to stable 2.4.0, add torchvision for compatibility
- Update Dockerfile to use specific ROCm PyTorch image version
This commit is contained in:
SlavaVlad
2026-05-13 02:15:48 +03:00
parent 67da25d849
commit 221ecbe69d
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

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

5
pyproject.toml
View File

@@ -19,8 +19,9 @@ dependencies = [
"av<16.0.0", "av<16.0.0",
"numpy>=2.1.0,<2.3.0; python_version >='3.13'", "numpy>=2.1.0,<2.3.0; python_version >='3.13'",
"pyannote-audio>=3.3.2,<4.0.0", "pyannote-audio>=3.3.2,<4.0.0",
"torch~=2.8.0", "torch~=2.4.0",
"torchaudio~=2.8.0", "torchaudio~=2.4.0",
"torchvision~=0.19.0",
"transformers>=4.48.0", "transformers>=4.48.0",
"triton>=3.3.0; sys_platform == 'linux' and platform_machine == 'x86_64'", # only install triton on x86_64 Linux "triton>=3.3.0; sys_platform == 'linux' and platform_machine == 'x86_64'", # only install triton on x86_64 Linux
"fastapi>=0.104.0", "fastapi>=0.104.0",