eland/requirements-dev.txt
Josh Devins 1ffbe002c4
Upgrade PyTorch dependencies to latest
In preparation for an 8.0 release, this updates PyTorch NLP dependencies
to more recent and latest minor versions. Amongst other things, this
introduces a fix from transformers that is helpful for text embedding
tasks with certain DPR models.

See: https://github.com/huggingface/transformers/issues/13670

Co-authored-by: Seth Michael Larson <seth.larson@elastic.co>
2021-12-06 09:05:54 -06:00

17 lines
223 B
Plaintext

elasticsearch>=7.7
pandas>=1.2.0
matplotlib
pytest>=5.2.1
pytest-mock
nbval
numpydoc>=0.9.0
scikit-learn>=0.22.1
xgboost>=1
nox
lightgbm
pytest-cov
mypy
sentence-transformers>=2.1.0
torch>=1.9.0
transformers[torch]>=4.12.0