transformers 4.46 compatibility

huggingface · Oct 23, 2024 · f0a24fc · f0a24fc
1 parent 227defe
commit f0a24fc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -28,7 +28,8 @@
 
 INSTALL_REQUIRE = [
     "torch>=1.11",
-    "transformers>=4.36,<4.46",
+    "transformers @ git+https://github.com/huggingface/transformers.git",
+    #"transformers>=4.36,<4.46",
     "optimum~=1.23",
     "datasets>=1.4.0",
     "sentencepiece",