Pin transformers>=4.35.0,<=4.38.2 (#408)

casper-hansen · Mar 24, 2024 · 0f94218 · 0f94218
1 parent 94e73f0
commit 0f94218
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -86,7 +86,7 @@ def get_kernels_whl_url(
 
 requirements = [
     "torch>=2.0.1",
-    "transformers>=4.35.0",
+    "transformers>=4.35.0,<=4.38.2",
     "tokenizers>=0.12.1",
     "typing_extensions>=4.8.0",
     "accelerate",