diff --git a/pyproject.toml b/pyproject.toml
index cde247a..d0977ff 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,11 +10,13 @@ readme = "README.md"
 python = "^3.10"
 pexpect = "^4.9.0"
 torch = "2.2.1"
-vllm = "0.4.1"
+# vllm = "0.4.1"
 openai = "1.30.5"
 fire = "0.6.0"
 wandb = "0.17.0"
 tenacity = "8.3.0"
+numpy = "1.26.4"
+
 
 [tool.poetry.build]
 generate-setup-file = false