Skip to content

GitLab

Unverified Commit b90a48f6 authored by Matt's avatar Matt Committed by GitHub
Browse files

Removing Keras version pinning (#14280) 

* Removing Keras version pinning

* make fixup
parent fd8136fa
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
setup.py
View file @ b90a48f6
......@@ -114,7 +114,6 @@ _deps = [
"jax>=0.2.8",
"jaxlib>=0.1.65",
"jieba",
"keras!=2.7.0", # Remove when they fix their release
"keras2onnx",
"nltk",
"numpy>=1.17",
......@@ -235,8 +234,8 @@ extras = {}
extras["ja"] = deps_list("fugashi", "ipadic", "unidic_lite", "unidic")
extras["sklearn"] = deps_list("scikit-learn")
extras["tf"] = deps_list("tensorflow", "onnxconverter-common", "keras2onnx", "keras")
extras["tf-cpu"] = deps_list("tensorflow-cpu", "onnxconverter-common", "keras2onnx", "keras")
extras["tf"] = deps_list("tensorflow", "onnxconverter-common", "keras2onnx")
extras["tf-cpu"] = deps_list("tensorflow-cpu", "onnxconverter-common", "keras2onnx")
extras["torch"] = deps_list("torch")
......
src/transformers/dependency_versions_table.py
View file @ b90a48f6
......@@ -25,7 +25,6 @@ deps = {
"jax": "jax>=0.2.8",
"jaxlib": "jaxlib>=0.1.65",
"jieba": "jieba",
"keras": "keras!=2.7.0",
"keras2onnx": "keras2onnx",
"nltk": "nltk",
"numpy": "numpy>=1.17",
......
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