accelerate>=0.29.2
bitsandbytes>=0.43.0
einops>=0.6.1
evaluate==0.4.0
datasets>=2.18.0
deepspeed>=0.14.4
hf_transfer>=0.1.4
huggingface-hub<1.0,>=0.19.2
jinja2>=3.0.0
ninja>=1.11.1
numpy>=1.24.2
packaging>=23.0
peft>=0.9.0
protobuf<=3.20.2
safetensors>=0.3.3
sentencepiece>=0.1.99
scipy
tensorboard
tqdm>=4.64.1
transformers>=4.39.3
trl>=0.9.6

[dev]
hf-doc-builder>=0.4.0
black>=24.4.2
isort>=5.12.0
flake8>=6.0.0
pytest
parameterized>=0.9.0

[docs]
hf-doc-builder>=0.4.0

[quality]
black>=24.4.2
isort>=5.12.0
flake8>=6.0.0

[tests]
pytest
parameterized>=0.9.0

[torch]
torch>=2.1.2
