From b90a48f6543d0adccf880f6076faec80c1d5c4c3 Mon Sep 17 00:00:00 2001 From: Matt Date: Thu, 4 Nov 2021 17:58:28 +0000 Subject: [PATCH] Removing Keras version pinning (#14280) * Removing Keras version pinning * make fixup --- setup.py | 5 ++--- src/transformers/dependency_versions_table.py | 1 - 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 9e4257ed5..0fb91a0cf 100644 --- a/setup.py +++ b/setup.py @@ -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") diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 2997951df..786e5a569 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -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",