Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
607615b7cd | ||
|
|
5daeb8f6ef |
@@ -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,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.
|
||||||
|
|||||||
@@ -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,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
|
||||||
@@ -15,4 +15,4 @@ setuptools-scm>=8
|
|||||||
runai-model-streamer[s3,gcs]==0.15.3
|
runai-model-streamer[s3,gcs]==0.15.3
|
||||||
conch-triton-kernels==1.2.1
|
conch-triton-kernels==1.2.1
|
||||||
timm>=1.0.17
|
timm>=1.0.17
|
||||||
grpcio-tools>=1.76.0
|
grpcio-tools>=1.76.0
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user