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

jmartin-tech · jmartin-tech · commit 0b837c1eb0b3 · 2024-12-09T15:02:55.000-06:00
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
