torch<=2.4.1,>=2.2.0
numpy<2.0
lightning==2.4.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

[all]
bitsandbytes==0.42.0
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.38.0
lm-eval>=0.4.2
huggingface_hub[hf_transfer]>=0.21.0

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

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