Skip to content

Commit

Permalink
Upgrade PyTorch dependencies to latest
Browse files Browse the repository at this point in the history
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: huggingface/transformers#13670

Co-authored-by: Seth Michael Larson <seth.larson@elastic.co>
  • Loading branch information
RayDev1988 and sethmlarson committed Dec 6, 2021
1 parent 325bd16 commit e312b58
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 6 deletions.
5 changes: 2 additions & 3 deletions requirements-dev.txt
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ nox
lightgbm
pytest-cov
mypy
huggingface-hub>=0.0.17
sentence-transformers>=2.0.0
sentence-transformers>=2.1.0
torch>=1.9.0
transformers[torch]>=4.11.0
transformers[torch]>=4.12.0
5 changes: 2 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,10 +59,9 @@
"scikit-learn": ["scikit-learn>=0.22.1,<1"],
"lightgbm": ["lightgbm>=2,<4"],
"pytorch": [
"huggingface-hub>=0.0.17,<1",
"sentence-transformers>=2.0.0,<3",
"sentence-transformers>=2.1.0,<3",
"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})
Expand Down

0 comments on commit e312b58

Please sign in to comment.