2024-04-05 00:39:17 -07:00
|
|
|
# Common dependencies
|
2025-03-08 17:44:35 +01:00
|
|
|
-r common.txt
|
2024-04-05 00:39:17 -07:00
|
|
|
|
2026-02-22 03:14:41 +08:00
|
|
|
# The version of gRPC libraries should be consistent with each other
|
|
|
|
|
grpcio==1.78.0
|
|
|
|
|
grpcio-reflection==1.78.0
|
|
|
|
|
grpcio-tools==1.78.0
|
|
|
|
|
|
2025-10-08 18:40:42 +01:00
|
|
|
numba == 0.61.2 # Required for N-gram speculative decoding
|
2025-03-05 09:55:00 +05:30
|
|
|
|
2024-04-05 00:39:17 -07:00
|
|
|
# Dependencies for AMD GPUs
|
2024-12-11 00:39:53 -08:00
|
|
|
datasets
|
2026-02-03 18:30:47 -05:00
|
|
|
ray[cgraph]>=2.48.0
|
2024-07-20 11:39:07 -05:00
|
|
|
peft
|
2024-05-20 13:29:28 -05:00
|
|
|
pytest-asyncio
|
2025-07-08 01:47:43 -04:00
|
|
|
tensorizer==2.10.1
|
2025-06-07 18:25:09 +05:30
|
|
|
packaging>=24.2
|
2025-11-12 19:46:58 -05:00
|
|
|
setuptools>=77.0.3,<80.0.0
|
2025-06-07 18:25:09 +05:30
|
|
|
setuptools-scm>=8
|
2025-12-04 12:04:43 +02:00
|
|
|
runai-model-streamer[s3,gcs]==0.15.3
|
2025-09-16 14:49:06 -05:00
|
|
|
conch-triton-kernels==1.2.1
|
2026-03-02 01:02:43 -05:00
|
|
|
timm>=1.0.17
|
|
|
|
|
# amd-quark: required for Quark quantization on ROCm
|
|
|
|
|
# To be consistent with test_quark.py
|
|
|
|
|
amd-quark>=0.8.99
|