From f7e990a69874aee9a82eefb51f094b73bf43905f Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Mon, 9 Mar 2026 18:01:39 +0100
Subject: [PATCH 1/9] Pin max Transformers version

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index f9ad971..69a3e18 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
     assert False, "Error: Could not open '%s' due %s\n" % (filepath, error)
 
 # ORT 1.16 is not compatible: https://github.com/Xilinx/Vitis-AI/issues/1343
-INSTALL_REQUIRE = ["optimum<2.0.0", "transformers>=4.38", "onnx", "onnxruntime-extensions"]
+INSTALL_REQUIRE = ["optimum<2.0.0", "transformers>=4.38,<=5.1.0", "onnx", "onnxruntime-extensions"]
 
 # TODO: unpin pytest once https://github.com/huggingface/transformers/pull/29154 is merged & released
 TESTS_REQUIRE = [

From 2a349b1ce9d4262dfe472ab9910ae28e7167ab23 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Mon, 9 Mar 2026 18:24:42 +0100
Subject: [PATCH 2/9] Test

---
 docs/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/Dockerfile b/docs/Dockerfile
index 15198a3..fecfab8 100644
--- a/docs/Dockerfile
+++ b/docs/Dockerfile
@@ -24,7 +24,7 @@ RUN pip install --no-cache-dir --upgrade pip uv
 ENV UV_SYSTEM_PYTHON=1
 RUN uv pip install --no-cache-dir git+https://github.com/huggingface/doc-builder.git
 
-RUN git clone $clone_url && cd optimum-amd && git checkout $commit_sha && uv pip install .[brevitas,tests]
+RUN git clone $clone_url && cd optimum-amd && git checkout fix_doc_build && uv pip install .[brevitas,tests]
 RUN git clone --depth 1 --branch v3.5 https://github.com/Xilinx/Vitis-AI.git && cd Vitis-AI/src/vai_quantizer/vai_q_onnx && sh build.sh && uv pip install pkgs/*.whl
 RUN uv pip uninstall torchao diffusers
 RUN uv pip install "onnxruntime==1.14"
\ No newline at end of file

From 0baa300a286d2ae32ce03e033ae2b70d348b3ed8 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Mon, 9 Mar 2026 18:41:14 +0100
Subject: [PATCH 3/9] Test

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 69a3e18..a78133f 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
     assert False, "Error: Could not open '%s' due %s\n" % (filepath, error)
 
 # ORT 1.16 is not compatible: https://github.com/Xilinx/Vitis-AI/issues/1343
-INSTALL_REQUIRE = ["optimum<2.0.0", "transformers>=4.38,<=5.1.0", "onnx", "onnxruntime-extensions"]
+INSTALL_REQUIRE = ["optimum<2.0.0", "transformers>=4.38,<=5.0.0", "onnx", "onnxruntime-extensions"]
 
 # TODO: unpin pytest once https://github.com/huggingface/transformers/pull/29154 is merged & released
 TESTS_REQUIRE = [

From 9b931f7c0ae4f8fb3069cc3e8c2fa9d5468b5753 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Mon, 9 Mar 2026 19:16:27 +0100
Subject: [PATCH 4/9] Test

---
 docs/Dockerfile | 2 +-
 setup.py        | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/docs/Dockerfile b/docs/Dockerfile
index fecfab8..76f797f 100644
--- a/docs/Dockerfile
+++ b/docs/Dockerfile
@@ -24,7 +24,7 @@ RUN pip install --no-cache-dir --upgrade pip uv
 ENV UV_SYSTEM_PYTHON=1
 RUN uv pip install --no-cache-dir git+https://github.com/huggingface/doc-builder.git
 
-RUN git clone $clone_url && cd optimum-amd && git checkout fix_doc_build && uv pip install .[brevitas,tests]
+RUN git clone $clone_url && cd optimum-amd && git checkout fix_doc_build && uv pip install .[brevitas,tests,doc]
 RUN git clone --depth 1 --branch v3.5 https://github.com/Xilinx/Vitis-AI.git && cd Vitis-AI/src/vai_quantizer/vai_q_onnx && sh build.sh && uv pip install pkgs/*.whl
 RUN uv pip uninstall torchao diffusers
 RUN uv pip install "onnxruntime==1.14"
\ No newline at end of file
diff --git a/setup.py b/setup.py
index a78133f..4e1840c 100644
--- a/setup.py
+++ b/setup.py
@@ -48,6 +48,7 @@
         "onnx-tool",
         "optimum>=1.17",
     ],
+    "doc": ["numpy<2"],
 }
 
 setup(

From 768e30b131d9cb20b526e994b2bded7369486494 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Tue, 10 Mar 2026 15:59:23 +0100
Subject: [PATCH 5/9] Unpin Optimum

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 4e1840c..a409a0c 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 from setuptools import find_namespace_packages, setup
 
 
-# Ensure we match the version set in optimum/intel/version.py
+# Ensure we match the version set in optimum/amd/version.py
 try:
     filepath = "optimum/amd/version.py"
     with open(filepath) as version_file:
@@ -14,7 +14,7 @@
     assert False, "Error: Could not open '%s' due %s\n" % (filepath, error)
 
 # ORT 1.16 is not compatible: https://github.com/Xilinx/Vitis-AI/issues/1343
-INSTALL_REQUIRE = ["optimum<2.0.0", "transformers>=4.38,<=5.0.0", "onnx", "onnxruntime-extensions"]
+INSTALL_REQUIRE = ["optimum", "transformers>=4.38,<=5.0.0", "onnx", "onnxruntime-extensions"]
 
 # TODO: unpin pytest once https://github.com/huggingface/transformers/pull/29154 is merged & released
 TESTS_REQUIRE = [

From ba336c0a0798664a9aefe56bd6bf0fafe75345b5 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Tue, 10 Mar 2026 16:40:40 +0100
Subject: [PATCH 6/9] Stricter pin for Transformers

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index a409a0c..f0e1e0d 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
     assert False, "Error: Could not open '%s' due %s\n" % (filepath, error)
 
 # ORT 1.16 is not compatible: https://github.com/Xilinx/Vitis-AI/issues/1343
-INSTALL_REQUIRE = ["optimum", "transformers>=4.38,<=5.0.0", "onnx", "onnxruntime-extensions"]
+INSTALL_REQUIRE = ["optimum<2.0.0", "transformers>=4.38,<5.0.0", "onnx", "onnxruntime-extensions"]
 
 # TODO: unpin pytest once https://github.com/huggingface/transformers/pull/29154 is merged & released
 TESTS_REQUIRE = [

From 46abd8bd140f9dfd2f9099e44b995077cb38e98f Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Tue, 10 Mar 2026 16:48:09 +0100
Subject: [PATCH 7/9] Remove Numpy pin?

---
 docs/Dockerfile | 2 +-
 setup.py        | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/docs/Dockerfile b/docs/Dockerfile
index 76f797f..fecfab8 100644
--- a/docs/Dockerfile
+++ b/docs/Dockerfile
@@ -24,7 +24,7 @@ RUN pip install --no-cache-dir --upgrade pip uv
 ENV UV_SYSTEM_PYTHON=1
 RUN uv pip install --no-cache-dir git+https://github.com/huggingface/doc-builder.git
 
-RUN git clone $clone_url && cd optimum-amd && git checkout fix_doc_build && uv pip install .[brevitas,tests,doc]
+RUN git clone $clone_url && cd optimum-amd && git checkout fix_doc_build && uv pip install .[brevitas,tests]
 RUN git clone --depth 1 --branch v3.5 https://github.com/Xilinx/Vitis-AI.git && cd Vitis-AI/src/vai_quantizer/vai_q_onnx && sh build.sh && uv pip install pkgs/*.whl
 RUN uv pip uninstall torchao diffusers
 RUN uv pip install "onnxruntime==1.14"
\ No newline at end of file
diff --git a/setup.py b/setup.py
index f0e1e0d..1c75310 100644
--- a/setup.py
+++ b/setup.py
@@ -48,7 +48,6 @@
         "onnx-tool",
         "optimum>=1.17",
     ],
-    "doc": ["numpy<2"],
 }
 
 setup(

From c68f9526cabe543d46bec800453927f25d07b1ea Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Tue, 10 Mar 2026 16:53:53 +0100
Subject: [PATCH 8/9] Pin Numpy

---
 docs/Dockerfile | 2 +-
 setup.py        | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/docs/Dockerfile b/docs/Dockerfile
index fecfab8..76f797f 100644
--- a/docs/Dockerfile
+++ b/docs/Dockerfile
@@ -24,7 +24,7 @@ RUN pip install --no-cache-dir --upgrade pip uv
 ENV UV_SYSTEM_PYTHON=1
 RUN uv pip install --no-cache-dir git+https://github.com/huggingface/doc-builder.git
 
-RUN git clone $clone_url && cd optimum-amd && git checkout fix_doc_build && uv pip install .[brevitas,tests]
+RUN git clone $clone_url && cd optimum-amd && git checkout fix_doc_build && uv pip install .[brevitas,tests,doc]
 RUN git clone --depth 1 --branch v3.5 https://github.com/Xilinx/Vitis-AI.git && cd Vitis-AI/src/vai_quantizer/vai_q_onnx && sh build.sh && uv pip install pkgs/*.whl
 RUN uv pip uninstall torchao diffusers
 RUN uv pip install "onnxruntime==1.14"
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 1c75310..f0e1e0d 100644
--- a/setup.py
+++ b/setup.py
@@ -48,6 +48,7 @@
         "onnx-tool",
         "optimum>=1.17",
     ],
+    "doc": ["numpy<2"],
 }
 
 setup(

From 5f50c030d896044410aa86b01c80b5b91419b1c1 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Tue, 10 Mar 2026 17:00:42 +0100
Subject: [PATCH 9/9] Clean

---
 docs/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/Dockerfile b/docs/Dockerfile
index 76f797f..3f2e95d 100644
--- a/docs/Dockerfile
+++ b/docs/Dockerfile
@@ -24,7 +24,7 @@ RUN pip install --no-cache-dir --upgrade pip uv
 ENV UV_SYSTEM_PYTHON=1
 RUN uv pip install --no-cache-dir git+https://github.com/huggingface/doc-builder.git
 
-RUN git clone $clone_url && cd optimum-amd && git checkout fix_doc_build && uv pip install .[brevitas,tests,doc]
+RUN git clone $clone_url && cd optimum-amd && git checkout $commit_sha && uv pip install .[brevitas,tests,doc]
 RUN git clone --depth 1 --branch v3.5 https://github.com/Xilinx/Vitis-AI.git && cd Vitis-AI/src/vai_quantizer/vai_q_onnx && sh build.sh && uv pip install pkgs/*.whl
 RUN uv pip uninstall torchao diffusers
 RUN uv pip install "onnxruntime==1.14"
\ No newline at end of file