diff --git a/setup.py b/setup.py
index cdb4e1b..8f78891 100644
--- a/setup.py
+++ b/setup.py
@@ -65,6 +65,7 @@ _deps = [
     "torch==2.1.0",
     "transformers==4.35.0",
     "trl==0.7.4",
+    "jinja2=>3.0.0",
     "tqdm>=4.64.1",
 ]