[Setup.py] update jaxlib (#9831)
* update jaxlib * Update setup.py * update table
This commit is contained in:
committed by
GitHub
parent
f617490e71
commit
d5b40d6693
18
setup.py
18
setup.py
@@ -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 = [
|
||||||
|
|||||||
@@ -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",
|
||||||
|
|||||||
Reference in New Issue
Block a user