diff --git a/pyproject.toml b/pyproject.toml
index cb537f73..8cc97ca4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -226,7 +226,7 @@ embeddings = [
 ml-heavy = [
     "torch==2.6.0",
     "torchaudio==2.6.0",
-    "numpy>=2.1,<2.2",
+    "numpy>=2.1,<2.3",
     "accelerate>=1.1.0",
     "bitsandbytes==0.44.1",
     "datasets>=3.0.2",