Преглед на файлове

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.
Alexander Borzunov преди 2 години
родител
ревизия
915b357740
променени са 2 файла, в които са добавени 2 реда и са изтрити 2 реда
  1. 1 1
      setup.cfg
  2. 1 1
      src/petals/__init__.py

+ 1 - 1
setup.cfg

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

+ 1 - 1
src/petals/__init__.py

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