diff --git a/docs/Dockerfile b/docs/Dockerfile
index 15198a3..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 $commit_sha && uv pip install .[brevitas,tests]
+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
diff --git a/setup.py b/setup.py
index f9ad971..f0e1e0d 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", "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 = [
@@ -48,6 +48,7 @@
         "onnx-tool",
         "optimum>=1.17",
     ],
+    "doc": ["numpy<2"],
 }
 
 setup(