Fix requiring transformers>=4.32.0 (#480)

pull/493/head
Alexander Borzunov 9 months ago committed by GitHub
parent 915b357740
commit 90840dfea2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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():

Loading…
Cancel
Save