pyyaml>=6.0

[dev]
pytest>=8.0

[train]
torch>=2.0
transformers>=4.40
datasets>=2.14
safetensors>=0.4
accelerate>=0.30
peft>=0.10
trl>=0.9
huggingface-hub>=0.23
bitsandbytes>=0.46.1
optuna>=3.0
pydantic>=2.0
pydantic-settings>=2.0
rich>=13.0
