accelerate==0.23.0
bitsandbytes==0.41.2.post2
einops>=0.6.1
evaluate==0.4.0
datasets==2.14.6
deepspeed==0.12.2
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.7.1
protobuf<=3.20.2
safetensors>=0.3.3
sentencepiece>=0.1.99
scipy
tensorboard
tqdm>=4.64.1
transformers==4.36.2
trl==0.7.10
matplotlib>=3.9.0
scikit-learn>=1.5.0

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

[docs]
hf-doc-builder>=0.4.0

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

[tests]
pytest
parameterized>=0.9.0

[torch]
torch==2.1.2
