diff --git a/amd_requirements.txt b/amd_requirements.txt index 7f2922d..940bb6c 100644 --- a/amd_requirements.txt +++ b/amd_requirements.txt @@ -1,7 +1,7 @@ # For AMD GPUs with DirectML (Windows) torch==2.5.1 torchvision==0.20.0 -torchaudio==2.5.0 +torchaudio==2.5.1 onnx==1.17.0 onnxruntime-directml==1.19.2 numpy diff --git a/nvidia_requirements.txt b/nvidia_requirements.txt index 6f80c9c..6e598cf 100644 --- a/nvidia_requirements.txt +++ b/nvidia_requirements.txt @@ -1,7 +1,7 @@ # For NVIDIA GPUs with CUDA support (Windows) torch==2.5.1 torchvision==0.20.0+cu118 -torchaudio==2.5.0+cu118 +torchaudio==2.5.1 onnx==1.17.0 onnxruntime-gpu==1.19.2 numpy # Required for PyTorch and ONNX operations