Skip to content

GitLab

Unverified Commit e5381899 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Upgrade jax/jaxlib/flax pin versions (#24791) 



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 6ba4d5de
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
setup.py
View file @ e5381899
...@@ -115,7 +115,7 @@ _deps = [ ...@@ -115,7 +115,7 @@ _deps = [
"faiss-cpu", "faiss-cpu",
"fastapi", "fastapi",
"filelock", "filelock",
"flax>=0.4.1,<=0.6.9", "flax>=0.4.1,<=0.7.0",
"ftfy", "ftfy",
"fugashi>=1.0", "fugashi>=1.0",
"GitPython<3.1.19", "GitPython<3.1.19",
...@@ -124,8 +124,8 @@ _deps = [ ...@@ -124,8 +124,8 @@ _deps = [
"importlib_metadata", "importlib_metadata",
"ipadic>=1.0.0,<2.0", "ipadic>=1.0.0,<2.0",
"isort>=5.5.4", "isort>=5.5.4",
"jax>=0.2.8,!=0.3.2,<=0.3.6", "jax>=0.2.8,!=0.3.2,<=0.4.13",
"jaxlib>=0.1.65,<=0.3.6", "jaxlib>=0.1.65,<=0.4.13",
"jieba", "jieba",
"kenlm", "kenlm",
"keras-nlp>=0.3.1", "keras-nlp>=0.3.1",
......
src/transformers/dependency_versions_table.py
View file @ e5381899
...@@ -20,7 +20,7 @@ deps = { ...@@ -20,7 +20,7 @@ deps = {
"faiss-cpu": "faiss-cpu", "faiss-cpu": "faiss-cpu",
"fastapi": "fastapi", "fastapi": "fastapi",
"filelock": "filelock", "filelock": "filelock",
"flax": "flax>=0.4.1,<=0.6.9", "flax": "flax>=0.4.1,<=0.7.0",
"ftfy": "ftfy", "ftfy": "ftfy",
"fugashi": "fugashi>=1.0", "fugashi": "fugashi>=1.0",
"GitPython": "GitPython<3.1.19", "GitPython": "GitPython<3.1.19",
...@@ -29,8 +29,8 @@ deps = { ...@@ -29,8 +29,8 @@ deps = {
"importlib_metadata": "importlib_metadata", "importlib_metadata": "importlib_metadata",
"ipadic": "ipadic>=1.0.0,<2.0", "ipadic": "ipadic>=1.0.0,<2.0",
"isort": "isort>=5.5.4", "isort": "isort>=5.5.4",
"jax": "jax>=0.2.8,!=0.3.2,<=0.3.6", "jax": "jax>=0.2.8,!=0.3.2,<=0.4.13",
"jaxlib": "jaxlib>=0.1.65,<=0.3.6", "jaxlib": "jaxlib>=0.1.65,<=0.4.13",
"jieba": "jieba", "jieba": "jieba",
"kenlm": "kenlm", "kenlm": "kenlm",
"keras-nlp": "keras-nlp>=0.3.1", "keras-nlp": "keras-nlp>=0.3.1",
......
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