diff --git a/src/petals/__init__.py b/src/petals/__init__.py index 5158464..404a481 100644 --- a/src/petals/__init__.py +++ b/src/petals/__init__.py @@ -17,7 +17,7 @@ __version__ = "2.1.0" 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.31.0,<5.0.0" + ), "Please install a proper transformers version: pip install transformers>=4.32.0,<5.0.0" def _override_bfloat16_mode_default():