[JAX] Bump min version (#25286)

* [JAX] Bump min version * make fixup

[JAX] Bump min version (#25286)
* [JAX] Bump min version * make fixup
66c240f3 · Sanchit Gandhi · GitHub · d114a6b7 · 66c240f3 · 66c240f3
Unverified Commit 66c240f3 authored Aug 03, 2023 by Sanchit Gandhi Committed by GitHub Aug 03, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

setup.py setup.py +2 -2

src/transformers/dependency_versions_table.py src/transformers/dependency_versions_table.py +2 -2

No files found.
--- a/setup.py
+++ b/setup.py
@@ -124,8 +124,8 @@ _deps = [
    "importlib_metadata",
    "ipadic>=1.0.0,<2.0",
    "isort>=5.5.4",
-    "jax>=0.2.8,!=0.3.2,<=0.4.13",
+    "jax>=0.4.1,<=0.4.13",
-    "jaxlib>=0.1.65,<=0.4.13",
+    "jaxlib>=0.4.1,<=0.4.13",
    "jieba",
    "kenlm",
    "keras-nlp>=0.3.1",

--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -29,8 +29,8 @@ deps = {
    "importlib_metadata": "importlib_metadata",
    "ipadic": "ipadic>=1.0.0,<2.0",
    "isort": "isort>=5.5.4",
-    "jax": "jax>=0.2.8,!=0.3.2,<=0.4.13",
+    "jax": "jax>=0.4.1,<=0.4.13",
-    "jaxlib": "jaxlib>=0.1.65,<=0.4.13",
+    "jaxlib": "jaxlib>=0.4.1,<=0.4.13",
    "jieba": "jieba",
    "kenlm": "kenlm",
    "keras-nlp": "keras-nlp>=0.3.1",