# Common dependencies
-r requirements-common.txt

# Dependencies for hcus
awscli
boto3
botocore
datasets
ray >= 2.10.0
peft
pytest-asyncio
tensorizer>=2.9.0
setuptools_scm>=8

torch == 2.4.1
triton == 3.0.0
flash_attn == 2.6.1
lmslim == 0.2.0  
numa