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 = [
"faiss-cpu",
"fastapi",
"filelock",
"flax>=0.4.1,<=0.6.9",
"flax>=0.4.1,<=0.7.0",
"ftfy",
"fugashi>=1.0",
"GitPython<3.1.19",
......@@ -124,8 +124,8 @@ _deps = [
"importlib_metadata",
"ipadic>=1.0.0,<2.0",
"isort>=5.5.4",
"jax>=0.2.8,!=0.3.2,<=0.3.6",
"jaxlib>=0.1.65,<=0.3.6",
"jax>=0.2.8,!=0.3.2,<=0.4.13",
"jaxlib>=0.1.65,<=0.4.13",
"jieba",
"kenlm",
"keras-nlp>=0.3.1",
......
src/transformers/dependency_versions_table.py
View file @ e5381899
......@@ -20,7 +20,7 @@ deps = {
"faiss-cpu": "faiss-cpu",
"fastapi": "fastapi",
"filelock": "filelock",
"flax": "flax>=0.4.1,<=0.6.9",
"flax": "flax>=0.4.1,<=0.7.0",
"ftfy": "ftfy",
"fugashi": "fugashi>=1.0",
"GitPython": "GitPython<3.1.19",
......@@ -29,8 +29,8 @@ deps = {
"importlib_metadata": "importlib_metadata",
"ipadic": "ipadic>=1.0.0,<2.0",
"isort": "isort>=5.5.4",
"jax": "jax>=0.2.8,!=0.3.2,<=0.3.6",
"jaxlib": "jaxlib>=0.1.65,<=0.3.6",
"jax": "jax>=0.2.8,!=0.3.2,<=0.4.13",
"jaxlib": "jaxlib>=0.1.65,<=0.4.13",
"jieba": "jieba",
"kenlm": "kenlm",
"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