Restrict transformers version until MPS issue is addressed (#1039)

NVIDIA · Dec 9, 2024 · 0b837c1 · 0b837c1
2 parents 7499da1 + 3120787
commit 0b837c1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -45,7 +45,7 @@ classifiers = [
 requires-python = ">=3.10"
 dependencies = [
   "base2048>=0.1.3",
-  "transformers>=4.43.0",
+  "transformers>=4.43.0,<4.47.0",
   "datasets>=2.14.6,<2.17",
   "colorama>=0.4.3",
   "tqdm>=4.64.0",

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 base2048>=0.1.3
-transformers>=4.43.0
+transformers>=4.43.0,<4.47.0
 datasets>=2.14.6,<2.17
 colorama>=0.4.3
 tqdm>=4.64.0
@@ -47,4 +47,4 @@ pytest_httpserver>=1.1.0
 black==24.4.2
 pylint>=3.1.0
 # calibration
-scipy>=1.14.0
+scipy>=1.14.0