From 970235a83801bb901e5cea96cfc28c05878f5b61 Mon Sep 17 00:00:00 2001
From: IlyasMoutawwakil <moutawwakil.ilyas.tsi@gmail.com>
Date: Thu, 24 Oct 2024 15:02:40 +0200
Subject: [PATCH] branch

---
 setup.py | 24 +++++++++++++++++++-----
 1 file changed, 19 insertions(+), 5 deletions(-)

diff --git a/setup.py b/setup.py
index 058cf6921f..9dad6f7142 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@
 REQUIRED_PKGS = [
     "coloredlogs",
     "sympy",
-    "transformers@git+https://github.com/huggingface/transformers.git@fix-pix2struct",
+    "transformers>=4.29,<4.47",
     "torch>=1.11",
     "packaging",
     "numpy",
@@ -54,7 +54,8 @@
         "datasets>=1.2.1",
         "evaluate",
         "protobuf>=3.20.1",
-        "transformers<4.47.0",
+        # "transformers<4.47.0",
+        "transformers@git+https://github.com/huggingface/transformers.git@fix-pix2struct",
     ],
     "onnxruntime-gpu": [
         "onnx",
@@ -63,10 +64,23 @@
         "evaluate",
         "protobuf>=3.20.1",
         "accelerate",  # ORTTrainer requires it.
-        "transformers<4.47.0",
+        # "transformers<4.47.0",
+        "transformers@git+https://github.com/huggingface/transformers.git@fix-pix2struct",
+    ],
+    "exporters": [
+        "onnx",
+        "onnxruntime",
+        "timm",
+        # "transformers<4.47.0",
+        "transformers@git+https://github.com/huggingface/transformers.git@fix-pix2struct",
+    ],
+    "exporters-gpu": [
+        "onnx",
+        "onnxruntime-gpu",
+        "timm",
+        # "transformers<4.47.0",
+        "transformers@git+https://github.com/huggingface/transformers.git@fix-pix2struct",
     ],
-    "exporters": ["onnx", "onnxruntime", "timm", "transformers<4.47.0"],
-    "exporters-gpu": ["onnx", "onnxruntime-gpu", "timm", "transformers<4.47.0"],
     "exporters-tf": [
         "tensorflow>=2.4,<=2.12.1",
         "tf2onnx",