torch>=2.0.0
transformers>=4.38.0
datasets>=2.14.0
numpy>=1.24.0
tqdm>=4.65.0
accelerate>=0.27.0

[all]
peft>=0.9.0
bitsandbytes>=0.42.0

[bitsandbytes]
bitsandbytes>=0.42.0

[dev]
pytest>=7.0
black
isort
mypy

[flash_attn]
flash-attn>=2.5.0

[peft]
peft>=0.9.0
