diff --git a/pyproject.toml b/pyproject.toml
index 23d524d..218569c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -28,7 +28,7 @@ dependencies = [
 [project.optional-dependencies]
 automl = [
   "flaml >= 2.0.0",
-  "xgboost >= 1.7.6",
+  "xgboost >= 1.7.6, < 2.0.0",
 ]
 embedding = [
   "transformers >= 4.30.2",
@@ -39,7 +39,7 @@ embedding = [
 ]
 all = [
   "flaml >= 2.0.0",
-  "xgboost >= 1.7.6",
+  "xgboost >= 1.7.6, < 2.0.0",
   "transformers >= 4.30.2",
   "sentence-transformers >= 2.2.1",
   "torch >= 2.0.1",