diff --git a/requirements/cuda.txt b/requirements/cuda.txt index 8e9a90638..378f84605 100644 --- a/requirements/cuda.txt +++ b/requirements/cuda.txt @@ -4,7 +4,7 @@ numba == 0.61.2 # Required for N-gram speculative decoding # Dependencies for NVIDIA GPUs -ray[cgraph]>=2.48.0 # Ray Compiled Graph, required for pipeline parallelism in V1. +ray[cgraph]>=2.48.0 torch==2.9.1 torchaudio==2.9.1 # These must be updated alongside torch diff --git a/requirements/rocm.txt b/requirements/rocm.txt index 5aeb16599..dc407995a 100644 --- a/requirements/rocm.txt +++ b/requirements/rocm.txt @@ -5,7 +5,7 @@ numba == 0.61.2 # Required for N-gram speculative decoding # Dependencies for AMD GPUs datasets -ray[cgraph]>=2.48.0 # Ray Compiled Graph, required for pipeline parallelism in V1. +ray[cgraph]>=2.48.0 peft pytest-asyncio tensorizer==2.10.1