diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 0e7217fb3769..44b56422e3ab 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -2,7 +2,7 @@ -r requirements-common.txt # Dependencies for NVIDIA GPUs -ray[default] >= 2.9 +ray[adag] == 2.41.0 # Required for pipeline parallelism in V1. torch == 2.5.1 torchaudio==2.5.1 # These must be updated alongside torch