# Common dependencies
-r common.txt

# The version of gRPC libraries should be consistent with each other
grpcio==1.78.0
grpcio-reflection==1.78.0

numba == 0.61.2 # Required for N-gram speculative decoding

# Dependencies for AMD GPUs
datasets
peft
pytest-asyncio
tensorizer==2.10.1
packaging>=24.2
setuptools>=77.0.3,<80.0.0
setuptools-scm>=8
runai-model-streamer[s3,gcs,azure]==0.15.7
# conch-triton-kernels==1.2.1
timm>=1.0.17
# amd-quark: required for Quark quantization on ROCm 
# To be consistent with test_quark.py
amd-quark>=0.8.99

# Other necessary dependencies
torch == 2.10.0
triton == 3.5.1
torchvision == 0.25.0
flash_attn == 2.8.3