[Setup.py] update jaxlib (#9831)

* update jaxlib

* Update setup.py

* update table
This commit is contained in:
Patrick von Platen
2021-01-27 09:34:21 +01:00
committed by GitHub
parent f617490e71
commit d5b40d6693
2 changed files with 17 additions and 5 deletions

View File

@@ -103,7 +103,7 @@ _deps = [
"ipadic>=1.0.0,<2.0", "ipadic>=1.0.0,<2.0",
"isort>=5.5.4", "isort>=5.5.4",
"jax>=0.2.0", "jax>=0.2.0",
"jaxlib==0.1.55", "jaxlib>=0.1.59",
"keras2onnx", "keras2onnx",
"numpy", "numpy",
"onnxconverter-common", "onnxconverter-common",
@@ -248,8 +248,20 @@ extras["dev"] = (
+ extras["modelcreation"] + extras["modelcreation"]
) )
extras["torchhub"] = deps_list("filelock", "importlib_metadata", "numpy", "packaging", "protobuf", "regex", extras["torchhub"] = deps_list(
"requests", "sacremoses", "sentencepiece", "torch", "tokenizers", "tqdm") "filelock",
"importlib_metadata",
"numpy",
"packaging",
"protobuf",
"regex",
"requests",
"sacremoses",
"sentencepiece",
"torch",
"tokenizers",
"tqdm",
)
# when modifying the following list, make sure to update src/transformers/dependency_versions_check.py # when modifying the following list, make sure to update src/transformers/dependency_versions_check.py
install_requires = [ install_requires = [

View File

@@ -16,7 +16,7 @@ deps = {
"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.0", "jax": "jax>=0.2.0",
"jaxlib": "jaxlib==0.1.55", "jaxlib": "jaxlib>=0.1.59",
"keras2onnx": "keras2onnx", "keras2onnx": "keras2onnx",
"numpy": "numpy", "numpy": "numpy",
"onnxconverter-common": "onnxconverter-common", "onnxconverter-common": "onnxconverter-common",