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
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
setup.py
View file @ e5f34871
......@@ -118,7 +118,7 @@ _deps = [
"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 @@ _deps = [
"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",
......
src/transformers/dependency_versions_table.py
View file @ e5f34871
......@@ -19,7 +19,7 @@ deps = {
"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 @@ deps = {
"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",
......
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