diff --git a/setup.py b/setup.py
index 44efd74..ddb775f 100644
--- a/setup.py
+++ b/setup.py
@@ -24,7 +24,7 @@
     name="py-tgi",
     version=PY_TGI_VERSION,
     packages=find_packages(),
-    install_requires=["docker", "huggingface-hub"],
+    install_requires=["docker", "huggingface-hub", "numpy"],
     extras_require={"quality": ["ruff"]},
     **common_setup_kwargs,
 )