Fix requiring transformers>=4.32.0 (#480)

pull/572/head
Alexander Borzunov 9 months ago committed by mike dupont
parent 1498e2936e
commit 44f4498883

@ -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