Pin flax & optax version (#22895)

* Pin optax version

* Pin flax too

* Fixup
This commit is contained in:
amyeroberts 2023-04-20 17:30:14 +01:00 committed by GitHub
parent 6dc0a849b7
commit e5f3487190
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions

View File

@ -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",

View File

@ -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",