SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Pin flax & optax version (#22895)

Browse Source 
* 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
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
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",