torch>=1.10.1
numpy>=1.21.2
pyyaml>=6.0
tqdm>=4.64.0
sentence-transformers>=2.2.2
transformers<4.31.0
torchmetrics>=0.11.4

[dev]
pytest==7.1.2
flake8==4.0.1
black==22.8.0
scikit-image==0.19.2
matplotlib==3.5.2
ipykernel==6.9.1
twine==4.0.1
