From dcce43670f79001f169a5d962d887ebb03530393 Mon Sep 17 00:00:00 2001 From: justheuristic Date: Tue, 7 Nov 2023 18:19:19 +0300 Subject: [PATCH] Hotfix: set transformers version <=4.34 temporarily (#538) * fix transformers version for now Co-authored-by: horik --- setup.cfg | 2 +- src/petals/__init__.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.cfg b/setup.cfg index c8dbc9a..e2be3d1 100644 --- a/setup.cfg +++ b/setup.cfg @@ -37,7 +37,7 @@ install_requires = accelerate>=0.22.0 huggingface-hub>=0.11.1,<1.0.0 tokenizers>=0.13.3 - transformers>=4.32.0,<5.0.0 # if you change this, please also change version assert in petals/__init__.py + transformers>=4.32.0,<4.35.0 # if you change this, please also change version assert in petals/__init__.py speedtest-cli==2.1.3 pydantic>=1.10,<2.0 # 2.0 is incompatible with hivemind yet hivemind==1.1.10.post2 diff --git a/src/petals/__init__.py b/src/petals/__init__.py index 8671fc2..e17b4a1 100644 --- a/src/petals/__init__.py +++ b/src/petals/__init__.py @@ -22,8 +22,8 @@ __version__ = "2.3.0.dev0" if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"): assert ( - version.parse("4.32.0") <= version.parse(transformers.__version__) < version.parse("5.0.0") - ), "Please install a proper transformers version: pip install transformers>=4.32.0,<5.0.0" + version.parse("4.32.0") <= version.parse(transformers.__version__) < version.parse("4.35.0") + ), "Please install a proper transformers version: pip install transformers>=4.32.0,<4.35.0" def _override_bfloat16_mode_default():