From 5819c1a84bc92e7f48b223c4c790ae17008258e9 Mon Sep 17 00:00:00 2001 From: justheuristic Date: Tue, 28 Feb 2023 17:47:32 +0300 Subject: [PATCH 1/2] Update dependency versions --- setup.cfg | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/setup.cfg b/setup.cfg index a05ae6b15..e88fe5592 100644 --- a/setup.cfg +++ b/setup.cfg @@ -32,13 +32,13 @@ packages = find: python_requires = >=3.7 install_requires = torch>=1.12 - bitsandbytes==0.34.0 - accelerate==0.15.0 + bitsandbytes==0.37.0 + accelerate==0.16.0 huggingface-hub==0.11.1 - transformers==4.25.1 + transformers==4.26.1 speedtest-cli==2.1.3 - hivemind==1.1.5 - tensor_parallel==1.0.23 + hivemind==1.1.6 + tensor_parallel==1.0.25 humanfriendly async-timeout>=4.0.2 cpufeature>=0.2.0 From 8bb8e86a6f5e61bb893826e62e1896ed98f96ed5 Mon Sep 17 00:00:00 2001 From: justheuristic Date: Tue, 28 Feb 2023 17:53:02 +0300 Subject: [PATCH 2/2] Update block.py --- src/petals/bloom/block.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/petals/bloom/block.py b/src/petals/bloom/block.py index f4d50be51..d420384c5 100644 --- a/src/petals/bloom/block.py +++ b/src/petals/bloom/block.py @@ -11,7 +11,7 @@ from transformers.models.bloom.modeling_bloom import BloomBlock, _expand_mask, _make_causal_mask, build_alibi_tensor if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"): - assert transformers.__version__.startswith("4.25."), "Please install transformers 4.25.1" + assert transformers.__version__.startswith("4.26."), "Please install transformers 4.26.1" class WrappedBloomBlock(BloomBlock):