diff --git a/requirements-dev.txt b/requirements-dev.txt index b8bcb0f..e424458 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -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' # diff --git a/setup.py b/setup.py index e9c8ecb..7d222e9 100644 --- a/setup.py +++ b/setup.py @@ -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", ], }