torch>=1.13.0
transformers>=4.21.0
datasets>=2.0.0
FlagEmbedding>=1.2.0
scikit-learn>=1.0.0
numpy==1.26.4
tqdm>=4.64.0
pathlib

[dev]
pytest>=7.0.0
pytest-cov>=3.0.0
black>=22.0.0
isort>=5.10.0
flake8>=4.0.0
mypy>=0.950

[training]
wandb>=0.12.0
tensorboard>=2.8.0
