third_codes/eland
1
0
Fork 0
mirror of https://github.com/elastic/eland.git synced 2025-07-11 00:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Upgrade PyTorch dependencies to latest

Browse Source 
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>
This commit is contained in:
Josh Devins 2021-12-06 16:05:54 +01:00 committed by GitHub
parent e6bb917d83
commit 1ffbe002c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
requirements-dev.txt
View File

@ -11,7 +11,6 @@ nox
lightgbm lightgbm
pytest-cov pytest-cov
mypy mypy
huggingface-hub>=0.0.17 sentence-transformers>=2.1.0
sentence-transformers>=2.0.0
torch>=1.9.0 torch>=1.9.0
transformers[torch]>=4.11.0 transformers[torch]>=4.12.0

5
setup.py
View File

@ -59,10 +59,9 @@ extras = {
"scikit-learn": ["scikit-learn>=0.22.1,<1"], "scikit-learn": ["scikit-learn>=0.22.1,<1"],
"lightgbm": ["lightgbm>=2,<4"], "lightgbm": ["lightgbm>=2,<4"],
"pytorch": [ "pytorch": [
"huggingface-hub>=0.0.17,<1", "sentence-transformers>=2.1.0,<3",
"sentence-transformers>=2.0.0,<3",
"torch>=1.9.0,<2", "torch>=1.9.0,<2",
"transformers[torch]>=4.11.0<5", "transformers[torch]>=4.12.0,<5",
], ],
} }
extras["all"] = list({dep for deps in extras.values() for dep in deps}) extras["all"] = list({dep for deps in extras.values() for dep in deps})