torch>=2.1.0
transformers>=4.36.0
tokenizers>=0.15.0
pyyaml>=6.0
tqdm>=4.66.0

[dev]
pytest>=7.4.0
black>=23.0
ruff>=0.1.0

[train]
datasets>=2.16.0
sentence-transformers>=2.2.0
accelerate>=0.25.0
wandb>=0.16.0
bert-score>=0.3.13
rouge-score>=0.1.2
scipy>=1.11.0
scikit-learn>=1.3.0
