Unverified Commit 5b143d33 authored by Nishidha's avatar Nishidha Committed by GitHub
Browse files

Moved numba from common requirements to cuda/rocm specific requirements (#14199)


Signed-off-by: default avatarNishidha Panpaliya <nishidha.panpaliya@partner.ibm.com>
parent eb59b5a6
psutil psutil
sentencepiece # Required for LLaMA tokenizer. sentencepiece # Required for LLaMA tokenizer.
numpy < 2.0.0 numpy < 2.0.0
numba == 0.60.0 # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding.
requests >= 2.26.0 requests >= 2.26.0
tqdm tqdm
blake3 blake3
......
# Common dependencies # Common dependencies
-r requirements-common.txt -r requirements-common.txt
numba == 0.60.0 # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding
# Dependencies for NVIDIA GPUs # Dependencies for NVIDIA GPUs
ray[cgraph] >= 2.43.0 # Ray Compiled Graph, required for pipeline parallelism in V1. ray[cgraph] >= 2.43.0 # Ray Compiled Graph, required for pipeline parallelism in V1.
torch == 2.5.1 torch == 2.5.1
......
# Common dependencies # Common dependencies
-r requirements-common.txt -r requirements-common.txt
numba == 0.60.0 # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding
# Dependencies for AMD GPUs # Dependencies for AMD GPUs
awscli awscli
boto3 boto3
...@@ -11,4 +13,4 @@ peft ...@@ -11,4 +13,4 @@ peft
pytest-asyncio pytest-asyncio
tensorizer>=2.9.0 tensorizer>=2.9.0
runai-model-streamer==0.11.0 runai-model-streamer==0.11.0
runai-model-streamer-s3==0.11.0 runai-model-streamer-s3==0.11.0
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment