diff --git a/requirements/cuda.txt b/requirements/cuda.txt index 9be7a868f56e..cdc6ee75afbc 100644 --- a/requirements/cuda.txt +++ b/requirements/cuda.txt @@ -2,7 +2,7 @@ -r common.txt numba == 0.60.0; python_version == '3.9' # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding -numba == 0.61; python_version > '3.9' +numba == 0.61.2; python_version > '3.9' # Dependencies for NVIDIA GPUs ray[cgraph]>=2.43.0, !=2.44.* # Ray Compiled Graph, required for pipeline parallelism in V1. diff --git a/requirements/rocm.txt b/requirements/rocm.txt index 5d5fea2d0e57..4df92aab3749 100644 --- a/requirements/rocm.txt +++ b/requirements/rocm.txt @@ -2,7 +2,7 @@ -r common.txt numba == 0.60.0; python_version == '3.9' # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding -numba == 0.61; python_version > '3.9' +numba == 0.61.2; python_version > '3.9' # Dependencies for AMD GPUs awscli diff --git a/requirements/test.in b/requirements/test.in index 9de492b8bbca..95c94dcdbe99 100644 --- a/requirements/test.in +++ b/requirements/test.in @@ -42,7 +42,7 @@ genai_perf==0.0.8 tritonclient==2.51.0 numba == 0.60.0; python_version == '3.9' # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding -numba == 0.61; python_version > '3.9' +numba == 0.61.2; python_version > '3.9' numpy runai-model-streamer==0.11.0 runai-model-streamer-s3==0.11.0 diff --git a/requirements/test.txt b/requirements/test.txt index 6fc5c2926170..476b4a2cc0ec 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -267,7 +267,7 @@ nltk==3.9.1 # via rouge-score num2words==0.5.14 # via -r requirements/test.in -numba==0.61.0 +numba==0.61.2 # via # -r requirements/test.in # librosa