diff --git a/requirements/cuda.txt b/requirements/cuda.txt index 9be7a868..cdc6ee75 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 5d5fea2d..4df92aab 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 9de492b8..95c94dcd 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 6fc5c292..476b4a2c 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