Hotfix: set transformers version <=4.34 temporarily (#538)

* fix transformers version for now


Co-authored-by: horik <hr.mail.qaq@gmail.com>
pull/539/head^2
justheuristic 6 months ago committed by GitHub
parent 82a97d6e9e
commit dcce43670f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

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

Loading…
Cancel
Save