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

Removing Keras version pinning (#14280)

Browse Source 
* Removing Keras version pinning

* make fixup
This commit is contained in:
Matt
2021-11-04 17:58:28 +00:00
committed by GitHub
parent fd8136fa75
commit b90a48f654
2 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
setup.py
View File

@@ -114,7 +114,6 @@ _deps = [
"jax>=0.2.8", "jax>=0.2.8",
"jaxlib>=0.1.65", "jaxlib>=0.1.65",
"jieba", "jieba",
"keras!=2.7.0", # Remove when they fix their release
"keras2onnx", "keras2onnx",
"nltk", "nltk",
"numpy>=1.17", "numpy>=1.17",
@@ -235,8 +234,8 @@ extras = {}
extras["ja"] = deps_list("fugashi", "ipadic", "unidic_lite", "unidic") extras["ja"] = deps_list("fugashi", "ipadic", "unidic_lite", "unidic")
extras["sklearn"] = deps_list("scikit-learn") extras["sklearn"] = deps_list("scikit-learn")
extras["tf"] = deps_list("tensorflow", "onnxconverter-common", "keras2onnx", "keras") extras["tf"] = deps_list("tensorflow", "onnxconverter-common", "keras2onnx")
extras["tf-cpu"] = deps_list("tensorflow-cpu", "onnxconverter-common", "keras2onnx", "keras") extras["tf-cpu"] = deps_list("tensorflow-cpu", "onnxconverter-common", "keras2onnx")
extras["torch"] = deps_list("torch") extras["torch"] = deps_list("torch")

1
src/transformers/dependency_versions_table.py
View File

@@ -25,7 +25,6 @@ deps = {
"jax": "jax>=0.2.8", "jax": "jax>=0.2.8",
"jaxlib": "jaxlib>=0.1.65", "jaxlib": "jaxlib>=0.1.65",
"jieba": "jieba", "jieba": "jieba",
"keras": "keras!=2.7.0",
"keras2onnx": "keras2onnx", "keras2onnx": "keras2onnx",
"nltk": "nltk", "nltk": "nltk",
"numpy": "numpy>=1.17", "numpy": "numpy>=1.17",