Support transformers 4.42

helena-intel · helena-intel · commit 5c077219cdc0 · 2024-06-28T11:12:26.000+02:00
diff --git a/setup.py b/setup.py
@@ -28,7 +28,7 @@
 
 INSTALL_REQUIRE = [
     "torch>=1.11",
-    "transformers>=4.36.0,<4.42.0",
+    "transformers>=4.36.0,<4.43.0",
     "optimum~=1.20",
     "datasets>=1.4.0",
     "sentencepiece",
