torch<2.6.0,>=2.5.0
numpy<2.0
lightning<2.6.0,>=2.5.0
jsonargparse[signatures]<=4.32.1,>=4.30.1
huggingface_hub>=0.23.5
safetensors>=0.4.3
tokenizers>=0.15.2
tqdm>=4.66.0

[:python_version >= "3.10" and sys_platform == "linux"]
lightning-thunder>=0.2.0.dev20250119

[all]
sentencepiece>=0.2.0
requests>=2.31.0
litdata==0.2.17
litserve<=0.2.4
zstandard>=0.22.0
pandas>=1.9.0
pyarrow>=15.0.2
tensorboard>=2.14.0
torchmetrics>=1.3.1
datasets>=2.18.0
transformers==4.47.1
lm-eval>=0.4.2
huggingface_hub[hf_transfer]>=0.21.0

[all:sys_platform != "win32"]
uvloop>=0.2.0

[all:sys_platform == "darwin"]
bitsandbytes<0.43.0,>=0.42.0

[all:sys_platform == "linux" or sys_platform == "win32"]
bitsandbytes<0.44.2,>=0.44.0

[test]
pytest>=8.1.1
pytest-rerunfailures>=14.0
pytest-timeout>=2.3.1
pytest-dependency>=0.6.0
transformers==4.47.1
einops>=0.7.0
protobuf>=4.23.4
