diff --git a/setup.py b/setup.py
index 8f78891..3c1c9e7 100644
--- a/setup.py
+++ b/setup.py
@@ -98,6 +98,7 @@ install_requires = [
     deps["datasets"],
     deps["deepspeed"],
     deps["huggingface-hub"],
+    deps["jinja2"],
     deps["ninja"],
     deps["numpy"],
     deps["packaging"],  # utilities from PyPA to e.g., compare versions