From 1ffbe002c4ca406bdd4fdba69c946660ae96a504 Mon Sep 17 00:00:00 2001 From: Josh Devins Date: Mon, 6 Dec 2021 16:05:54 +0100 Subject: [PATCH] Upgrade PyTorch dependencies to latest 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 --- requirements-dev.txt | 5 ++--- setup.py | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/requirements-dev.txt b/requirements-dev.txt index a074f42..a2c977c 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -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 diff --git a/setup.py b/setup.py index 4f0f2c5..4bcbe52 100644 --- a/setup.py +++ b/setup.py @@ -59,10 +59,9 @@ extras = { "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})