Skip to content

GitLab

Unverified Commit e5f34871 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Pin flax & optax version (#22895) 

* Pin optax version

* Pin flax too

* Fixup
parent 6dc0a849
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
setup.py
View file @ e5f34871
...@@ -118,7 +118,7 @@ _deps = [ ...@@ -118,7 +118,7 @@ _deps = [
"faiss-cpu", "faiss-cpu",
"fastapi", "fastapi",
"filelock", "filelock",
"flax>=0.4.1", "flax>=0.4.1,<=0.6.9",
"ftfy", "ftfy",
"fugashi>=1.0", "fugashi>=1.0",
"GitPython<3.1.19", "GitPython<3.1.19",
...@@ -140,7 +140,7 @@ _deps = [ ...@@ -140,7 +140,7 @@ _deps = [
"onnxruntime-tools>=1.4.2", "onnxruntime-tools>=1.4.2",
"onnxruntime>=1.4.0", "onnxruntime>=1.4.0",
"optuna", "optuna",
"optax>=0.0.8", "optax>=0.0.8,<=0.1.4",
"packaging>=20.0", "packaging>=20.0",
"parameterized", "parameterized",
"phonemizer", "phonemizer",
......
src/transformers/dependency_versions_table.py
View file @ e5f34871
...@@ -19,7 +19,7 @@ deps = { ...@@ -19,7 +19,7 @@ deps = {
"faiss-cpu": "faiss-cpu", "faiss-cpu": "faiss-cpu",
"fastapi": "fastapi", "fastapi": "fastapi",
"filelock": "filelock", "filelock": "filelock",
"flax": "flax>=0.4.1", "flax": "flax>=0.4.1,<=0.6.9",
"ftfy": "ftfy", "ftfy": "ftfy",
"fugashi": "fugashi>=1.0", "fugashi": "fugashi>=1.0",
"GitPython": "GitPython<3.1.19", "GitPython": "GitPython<3.1.19",
...@@ -41,7 +41,7 @@ deps = { ...@@ -41,7 +41,7 @@ deps = {
"onnxruntime-tools": "onnxruntime-tools>=1.4.2", "onnxruntime-tools": "onnxruntime-tools>=1.4.2",
"onnxruntime": "onnxruntime>=1.4.0", "onnxruntime": "onnxruntime>=1.4.0",
"optuna": "optuna", "optuna": "optuna",
"optax": "optax>=0.0.8", "optax": "optax>=0.0.8,<=0.1.4",
"packaging": "packaging>=20.0", "packaging": "packaging>=20.0",
"parameterized": "parameterized", "parameterized": "parameterized",
"phonemizer": "phonemizer", "phonemizer": "phonemizer",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment