瀏覽代碼

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

* fix transformers version for now


Co-authored-by: horik <hr.mail.qaq@gmail.com>
justheuristic 1 年之前
父節點
當前提交
dcce43670f
共有 2 個文件被更改,包括 3 次插入3 次删除
  1. 1 1
      setup.cfg
  2. 2 2
      src/petals/__init__.py

+ 1 - 1
setup.cfg

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

+ 2 - 2
src/petals/__init__.py

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