diff --git a/pyproject.toml b/pyproject.toml
index 90df64ad2ae33..bcedbb53ab887 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,6 +7,7 @@ requires = [
     "setuptools >= 49.4.0",
     "torch == 2.4.0",
     "wheel",
+    "jinja2",
 ]
 build-backend = "setuptools.build_meta"