mirror of
https://github.com/elastic/eland.git
synced 2025-07-11 00:02:14 +08:00
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>
17 lines
223 B
Plaintext
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
|