
[all]
torch>=1.9.0
transformers>=4.0.0
sentence-transformers>=2.0.0

[dev]
pytest>=6.0
pytest-cov
black
flake8
mypy

[gpu]
torch>=1.9.0

[sentence-transformers]
sentence-transformers>=2.0.0

[transformers]
transformers>=4.0.0
