Bump minimum PyTorch version to 1.11

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

View File

@ -17,7 +17,7 @@ lightgbm>=2,<4
# PyTorch doesn't support Python 3.10 yet (pytorch/pytorch#66424)
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'
#

View File

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