torch>=1.10
numpy
tqdm
pyyaml
scipy
wandb
rich>=13.0.0
requests>=2.28
transformers>=4.40
peft>=0.10
trl>=0.8
datasets>=2.14
accelerate>=0.28

[:sys_platform == "linux" or sys_platform == "win32"]
bitsandbytes>=0.43

[all]

[all:sys_platform == "linux"]
flash-attn>=2.3

[dev]
pytest>=7.0
pytest-cov
black
flake8
mypy
isort
