Require transformers>=4.32.0 (#479)

It's necessary to load https://huggingface.co/petals-team/StableBeluga2 since it doesn't have deprecated `inv_freq` weights.
pull/480/head
Alexander Borzunov 9 months ago committed by GitHub
parent 18e93afc73
commit 915b357740
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -36,7 +36,7 @@ install_requires =
accelerate>=0.22.0
huggingface-hub>=0.11.1,<1.0.0
tokenizers>=0.13.3
transformers>=4.31.0,<5.0.0 # if you change this, please also change version assert in petals/__init__.py
transformers>=4.32.0,<5.0.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.9

@ -16,7 +16,7 @@ __version__ = "2.1.0"
if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"):
assert (
version.parse("4.31.0") <= version.parse(transformers.__version__) < version.parse("5.0.0")
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"

Loading…
Cancel
Save