vllm/requirements-rocm.txt
Nishidha 5b143d33be
Moved numba from common requirements to cuda/rocm specific requirements (#14199)
Signed-off-by: Nishidha Panpaliya <nishidha.panpaliya@partner.ibm.com>
2025-03-05 12:25:00 +08:00

17 lines
316 B
Plaintext

# Common dependencies
-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
awscli
boto3
botocore
datasets
ray >= 2.10.0
peft
pytest-asyncio
tensorizer>=2.9.0
runai-model-streamer==0.11.0
runai-model-streamer-s3==0.11.0