revert

Author: echarlaix

setup.py

@@ -13,8 +13,7 @@
 
 
 REQUIRED_PKGS = [
-    "transformers@git+https://github.com/echarlaix/transformers.git@int32-dtype",
-    # "transformers>=4.29",
+    "transformers>=4.29",
     "torch>=1.11",
     "packaging",
     "numpy",
@@ -51,15 +50,15 @@
         "datasets>=1.2.1",
         "evaluate",
         "protobuf>=3.20.1",
-        "transformers>=4.36,<4.51.0", # TODO: remove
+        "transformers>=4.36,<4.50.0",
     ],
     "onnxruntime-gpu": [
         "onnx",
         "onnxruntime-gpu>=1.11.0",
         "datasets>=1.2.1",
         "evaluate",
         "protobuf>=3.20.1",
-        "transformers>=4.36,<4.51.0",
+        "transformers>=4.36,<4.50.0",
     ],
     "onnxruntime-training": [
         "torch-ort",
@@ -68,19 +67,19 @@
         "accelerate",
         "evaluate",
         "protobuf>=3.20.1",
-        "transformers>=4.36,<4.51.0",
+        "transformers>=4.36,<4.50.0",
     ],
     "exporters": [
         "onnx",
         "onnxruntime",
         "timm",
-        "transformers>=4.36,<4.51.0",
+        "transformers>=4.36,<4.50.0",
     ],
     "exporters-gpu": [
         "onnx",
         "onnxruntime-gpu",
         "timm",
-        "transformers>=4.36,<4.51.0",
+        "transformers>=4.36,<4.50.0",
     ],
     "exporters-tf": [
         "tensorflow>=2.4,<=2.12.1",