diff --git a/setup.py b/setup.py index 4edffc724e9..55dc4fa353c 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 3148d0f3393..d699a6ac7f6 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",