packaging>=20.9
wheel>=0.43
pyyaml
py-cpuinfo
torch>=2.3.0a0
transformers>=4.41.2
accelerate>=0.34.2
datasets>=2.15.0
numba
rich
instructlab-dolomite>=0.1.1
trl>=0.9.4
peft
pydantic>=2.7.0
deepspeed>=0.14.3
aiofiles>=23.2.1

[:python_version != "3.10"]
numpy<2.0.0,>=1.26.4

[:python_version == "3.10"]
numpy<2.0.0,>=1.23.5

[cuda]
flash-attn>=2.4.0
bitsandbytes>=0.43.1
