Pin flax & optax version (huggingface#22895)

* Pin optax version * Pin flax too * Fixup
novice03 · Jun 23, 2023 · 2a40388 · 2a40388
1 parent 718211f
commit 2a40388
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -118,7 +118,7 @@
     "faiss-cpu",
     "fastapi",
     "filelock",
-    "flax>=0.4.1",
+    "flax>=0.4.1,<=0.6.9",
     "ftfy",
     "fugashi>=1.0",
     "GitPython<3.1.19",
@@ -140,7 +140,7 @@
     "onnxruntime-tools>=1.4.2",
     "onnxruntime>=1.4.0",
     "optuna",
-    "optax>=0.0.8",
+    "optax>=0.0.8,<=0.1.4",
     "packaging>=20.0",
     "parameterized",
     "phonemizer",

diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
@@ -19,7 +19,7 @@
     "faiss-cpu": "faiss-cpu",
     "fastapi": "fastapi",
     "filelock": "filelock",
-    "flax": "flax>=0.4.1",
+    "flax": "flax>=0.4.1,<=0.6.9",
     "ftfy": "ftfy",
     "fugashi": "fugashi>=1.0",
     "GitPython": "GitPython<3.1.19",
@@ -41,7 +41,7 @@
     "onnxruntime-tools": "onnxruntime-tools>=1.4.2",
     "onnxruntime": "onnxruntime>=1.4.0",
     "optuna": "optuna",
-    "optax": "optax>=0.0.8",
+    "optax": "optax>=0.0.8,<=0.1.4",
     "packaging": "packaging>=20.0",
     "parameterized": "parameterized",
     "phonemizer": "phonemizer",