Skip to content

Commit dcce436

Browse files
justheuristichrQAQ
andauthored
Hotfix: set transformers version <=4.34 temporarily (#538)
* fix transformers version for now Co-authored-by: horik <[email protected]>
1 parent 82a97d6 commit dcce436

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

setup.cfg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ install_requires =
3737
accelerate>=0.22.0
3838
huggingface-hub>=0.11.1,<1.0.0
3939
tokenizers>=0.13.3
40-
transformers>=4.32.0,<5.0.0 # if you change this, please also change version assert in petals/__init__.py
40+
transformers>=4.32.0,<4.35.0 # if you change this, please also change version assert in petals/__init__.py
4141
speedtest-cli==2.1.3
4242
pydantic>=1.10,<2.0 # 2.0 is incompatible with hivemind yet
4343
hivemind==1.1.10.post2

src/petals/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@
2222

2323
if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"):
2424
assert (
25-
version.parse("4.32.0") <= version.parse(transformers.__version__) < version.parse("5.0.0")
26-
), "Please install a proper transformers version: pip install transformers>=4.32.0,<5.0.0"
25+
version.parse("4.32.0") <= version.parse(transformers.__version__) < version.parse("4.35.0")
26+
), "Please install a proper transformers version: pip install transformers>=4.32.0,<4.35.0"
2727

2828

2929
def _override_bfloat16_mode_default():

0 commit comments

Comments
 (0)