From ec1f9c8c9160d041c2ad17d60197d1db8adb8b22 Mon Sep 17 00:00:00 2001 From: cyyever Date: Thu, 10 Apr 2025 15:59:37 +0800 Subject: [PATCH] Update Numba to 0.61.2 (#16376) Signed-off-by: cyy --- requirements/cuda.txt | 2 +- requirements/rocm.txt | 2 +- requirements/test.in | 2 +- requirements/test.txt | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) 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