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

Bump minimum PyTorch version to 1.11

Browse Source
This commit is contained in:
David Kyle 2022-06-21 13:43:43 +01:00 committed by GitHub
parent 081c8efaa0
commit 8448b3ba4e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements-dev.txt
View File

@ -17,7 +17,7 @@ lightgbm>=2,<4
# PyTorch doesn't support Python 3.10 yet (pytorch/pytorch#66424) # PyTorch doesn't support Python 3.10 yet (pytorch/pytorch#66424)
sentence-transformers>=2.1.0,<3; python_version<'3.10' sentence-transformers>=2.1.0,<3; python_version<'3.10'
torch>=1.9.0,<2; python_version<'3.10' torch>=1.11.0,<2; python_version<'3.10'
transformers[torch]>=4.12.0,<5; python_version<'3.10' transformers[torch]>=4.12.0,<5; python_version<'3.10'
# #

2
setup.py
View File

@ -60,7 +60,7 @@ extras = {
"lightgbm": ["lightgbm>=2,<4"], "lightgbm": ["lightgbm>=2,<4"],
"pytorch": [ "pytorch": [
"sentence-transformers>=2.1.0,<3", "sentence-transformers>=2.1.0,<3",
"torch>=1.9.0,<2", "torch>=1.11.0,<2",
"transformers[torch]>=4.12.0,<5", "transformers[torch]>=4.12.0,<5",
], ],
} }