From 98ad88574d48f49192a8462463e65f347e4d24ba Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Thu, 2 May 2024 09:11:15 +0200 Subject: [PATCH] please install the correct keras version --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index eff7ee32314954..a09ae26840f042 100644 --- a/setup.py +++ b/setup.py @@ -258,7 +258,7 @@ def run(self): extras["sklearn"] = deps_list("scikit-learn") extras["tf"] = deps_list("tensorflow", "onnxconverter-common", "tf2onnx", "tensorflow-text", "keras-nlp") -extras["tf-cpu"] = deps_list("tensorflow-cpu", "onnxconverter-common", "tf2onnx", "keras", "tensorflow-text", "keras-nlp") +extras["tf-cpu"] = deps_list("keras", "tensorflow-cpu", "onnxconverter-common", "tf2onnx", "tensorflow-text", "keras-nlp") extras["torch"] = deps_list("torch", "accelerate") extras["accelerate"] = deps_list("accelerate")