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

[JAX] Bump jax lib (#12053)

Browse Source 
* fix_torch_device_generate_test

* remove @

* bump up jax lib
This commit is contained in:
Patrick von Platen
2021-06-07 13:04:18 +01:00
committed by GitHub
parent 185122ef22
commit 59f75d538b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -104,7 +104,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.8", "jax>=0.2.8",
"jaxlib>=0.1.59", "jaxlib>=0.1.65",
"jieba", "jieba",
"keras2onnx", "keras2onnx",
"nltk", "nltk",