Compare commits

...
Author SHA1 Message Date
Tyler Michael Smith 607615b7cd pre commit, typo fix
Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com>
2026-01-27 09:38:45 -05:00
Tyler Michael Smith 5daeb8f6ef Upgrade Numba to fix segfault on CUDA drivers version 580
Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com>
2026-01-27 09:20:29 -05:00
7 changed files with 9 additions and 9 deletions
+1 -1
View File
@@ -3,7 +3,7 @@
setuptools==77.0.3 # this version can reuse CMake build dir 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
# Dependencies for CPUs # Dependencies for CPUs
torch==2.10.0+cpu; platform_machine == "x86_64" or platform_machine == "s390x" torch==2.10.0+cpu; platform_machine == "x86_64" or platform_machine == "s390x"
+1 -1
View File
@@ -1,7 +1,7 @@
# Common dependencies # Common dependencies
-r common.txt -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 # Dependencies for NVIDIA GPUs
ray[cgraph]>=2.48.0 # Ray Compiled Graph, required for pipeline parallelism in V1. ray[cgraph]>=2.48.0 # Ray Compiled Graph, required for pipeline parallelism in V1.
+1 -1
View File
@@ -40,7 +40,7 @@ buildkite-test-collector==0.1.9
genai_perf>=0.0.8 genai_perf>=0.0.8
tritonclient>=2.51.0 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 numpy
runai-model-streamer[s3,gcs]==0.15.3 runai-model-streamer[s3,gcs]==0.15.3
fastsafetensors>=0.1.10 fastsafetensors>=0.1.10
+1 -1
View File
@@ -1,7 +1,7 @@
# Common dependencies # Common dependencies
-r common.txt -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 AMD GPUs # Dependencies for AMD GPUs
datasets datasets
+1 -1
View File
@@ -49,7 +49,7 @@ genai_perf>=0.0.8
tritonclient>=2.51.0 tritonclient>=2.51.0
arctic-inference == 0.1.1 # Required for suffix decoding test 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 numpy
runai-model-streamer[s3,gcs]==0.15.3 runai-model-streamer[s3,gcs]==0.15.3
fastsafetensors>=0.1.10 fastsafetensors>=0.1.10
+2 -2
View File
@@ -456,7 +456,7 @@ lightning-utilities==0.14.3
# lightning # lightning
# pytorch-lightning # pytorch-lightning
# torchmetrics # torchmetrics
llvmlite==0.44.0 llvmlite==0.46.0
# via numba # via numba
lm-eval==0.4.9.2 lm-eval==0.4.9.2
# via -r requirements/test.in # via -r requirements/test.in
@@ -527,7 +527,7 @@ nltk==3.9.1
# via rouge-score # via rouge-score
num2words==0.5.14 num2words==0.5.14
# via -r requirements/test.in # via -r requirements/test.in
numba==0.61.2 numba==0.63.1
# via # via
# -r requirements/test.in # -r requirements/test.in
# librosa # librosa
+1 -1
View File
@@ -9,7 +9,7 @@ setuptools>=77.0.3,<81.0.0
wheel wheel
jinja2>=3.1.6 jinja2>=3.1.6
datasets # for benchmark scripts 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 --extra-index-url=https://download.pytorch.org/whl/xpu
torch==2.9.0+xpu torch==2.9.0+xpu
torchaudio torchaudio