Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion requirements/cpu.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

setuptools==77.0.3 # this version can reuse CMake build dir

numba == 0.61.2; platform_machine != "s390x" # Required for N-gram speculative decoding
numba == 0.63.1; platform_machine != "s390x" # Required for N-gram speculative decoding
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

The stated purpose of this pull request is to add support for Python 3.14 by upgrading numba. However, numba version 0.63.1 does not provide support for Python 3.14; it introduces initial support for Python 3.13. This means the current change does not achieve the stated goal. A newer version of numba will be necessary to support Python 3.14 when it becomes available.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

See the release notes, 3.14 support is explicitly noted.


# Dependencies for CPUs
torch==2.11.0+cpu; platform_machine == "x86_64" or platform_machine == "s390x" or platform_machine == "aarch64"
Expand Down
2 changes: 1 addition & 1 deletion requirements/cuda.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# Common dependencies
-r common.txt

numba == 0.61.2 # Required for N-gram speculative decoding
numba == 0.63.1 # Required for N-gram speculative decoding

# Dependencies for NVIDIA GPUs
torch==2.11.0
Expand Down
2 changes: 1 addition & 1 deletion requirements/nightly_torch_test.txt
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ buildkite-test-collector==0.1.9
genai_perf>=0.0.8
tritonclient>=2.51.0

numba == 0.61.2 # Required for N-gram speculative decoding
numba == 0.63.1 # Required for N-gram speculative decoding
numpy
runai-model-streamer[s3,gcs,azure]==0.15.7
fastsafetensors>=0.2.2
Expand Down
2 changes: 1 addition & 1 deletion requirements/rocm.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
grpcio==1.78.0
grpcio-reflection==1.78.0

numba == 0.61.2 # Required for N-gram speculative decoding
numba == 0.63.1 # Required for N-gram speculative decoding

# Dependencies for AMD GPUs
datasets
Expand Down
2 changes: 1 addition & 1 deletion requirements/test.in
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ grpcio==1.78.0
grpcio-reflection==1.78.0

arctic-inference == 0.1.1 # Required for suffix decoding test
numba == 0.61.2 # Required for N-gram speculative decoding
numba == 0.63.1 # Required for N-gram speculative decoding
numpy
runai-model-streamer[s3,gcs,azure]==0.15.7
fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage
Expand Down
4 changes: 2 additions & 2 deletions requirements/test.txt
Original file line number Diff line number Diff line change
Expand Up @@ -477,7 +477,7 @@ lightning-utilities==0.14.3
# lightning
# pytorch-lightning
# torchmetrics
llvmlite==0.44.0
llvmlite==0.46.0
# via numba
lm-eval==0.4.11
# via -r requirements/test.in
Expand Down Expand Up @@ -548,7 +548,7 @@ nltk==3.9.1
# via rouge-score
num2words==0.5.14
# via -r requirements/test.in
numba==0.61.2
numba==0.63.1
# via
# -r requirements/test.in
# librosa
Expand Down
2 changes: 1 addition & 1 deletion requirements/xpu.txt
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ setuptools>=77.0.3,<81.0.0
wheel
jinja2>=3.1.6
datasets # for benchmark scripts
numba == 0.61.2 # Required for N-gram speculative decoding
numba == 0.63.1 # Required for N-gram speculative decoding
--extra-index-url=https://download.pytorch.org/whl/xpu
torch==2.11.0+xpu
torchaudio
Expand Down
Loading