From b1b0c1c8c0c4d247bd5ab9562947e53e7100507f Mon Sep 17 00:00:00 2001
From: lewtun <lewis.c.tunstall@gmail.com>
Date: Fri, 10 Nov 2023 10:44:06 +0100
Subject: [PATCH] Update setup.py

Co-authored-by: Philipp Schmid <32632186+philschmid@users.noreply.github.com>
---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

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",
 ]