[Flax] Bump to v0.4.1 (#17966)

2022-07-05 15:17:17 +01:00
parent 97db5b4223
commit ec07eccc7d
2 changed files with 2 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -110,7 +110,7 @@ _deps = [
    "fastapi",
    "filelock",
    "flake8>=3.8.3",
-    "flax>=0.3.5",
+    "flax>=0.4.1",
    "ftfy",
    "fugashi>=1.0",
    "GitPython<3.1.19",
--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -16,7 +16,7 @@ deps = {
    "fastapi": "fastapi",
    "filelock": "filelock",
    "flake8": "flake8>=3.8.3",
-    "flax": "flax>=0.3.5",
+    "flax": "flax>=0.4.1",
    "ftfy": "ftfy",
    "fugashi": "fugashi>=1.0",
    "GitPython": "GitPython<3.1.19",