torch>=2.1.0
transformers>=4.40.0
accelerate>=0.27.0

[dev]
pytest>=7.0
black
ruff
mypy

[full]
torch>=2.1.0
transformers>=4.40.0
accelerate>=0.27.0
bitsandbytes>=0.43.0
peft>=0.10.0
trl>=0.8.0
datasets>=2.18.0
triton>=2.1.0
deepspeed>=0.14.0
optuna>=3.5.0
flash-attn>=2.5.0

[train]
bitsandbytes>=0.43.0
peft>=0.10.0
trl>=0.8.0
datasets>=2.18.0
optuna>=3.5.0
