From 2d4820284d823ad362e63662397aae957a95901b Mon Sep 17 00:00:00 2001 From: Matt Date: Mon, 24 Jun 2024 14:42:16 +0100 Subject: [PATCH] Add Jinja as a requirement with the right version cutoff (#31536) * Add Jinja as a requirement with the right version cutoff * Correct package name! --- setup.py | 1 + src/transformers/dependency_versions_table.py | 1 + 2 files changed, 2 insertions(+) diff --git a/setup.py b/setup.py index 4edffc724e..55dc4fa353 100644 --- a/setup.py +++ b/setup.py @@ -124,6 +124,7 @@ _deps = [ "jax>=0.4.1,<=0.4.13", "jaxlib>=0.4.1,<=0.4.13", "jieba", + "jinja2>=3.1.0", "kenlm", # Keras pin - this is to make sure Keras 3 doesn't destroy us. Remove or change when we have proper support. "keras>2.9,<2.16", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 3148d0f339..d699a6ac7f 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -31,6 +31,7 @@ deps = { "jax": "jax>=0.4.1,<=0.4.13", "jaxlib": "jaxlib>=0.4.1,<=0.4.13", "jieba": "jieba", + "jinja2": "jinja2>=3.1.0", "kenlm": "kenlm", "keras": "keras>2.9,<2.16", "keras-nlp": "keras-nlp>=0.3.1",