Also pin TensorFlow CPU

This commit is contained in:
Sylvain Gugger
2022-11-18 08:50:56 -05:00
parent a3f7458066
commit 37e016331f
2 changed files with 2 additions and 2 deletions

View File

@@ -156,7 +156,7 @@ _deps = [
"sigopt", "sigopt",
"librosa", "librosa",
"starlette", "starlette",
"tensorflow-cpu>=2.3", "tensorflow-cpu>=2.4,<2.11",
"tensorflow>=2.4,<2.11", "tensorflow>=2.4,<2.11",
"tensorflow-text", "tensorflow-text",
"tf2onnx", "tf2onnx",

View File

@@ -62,7 +62,7 @@ deps = {
"sigopt": "sigopt", "sigopt": "sigopt",
"librosa": "librosa", "librosa": "librosa",
"starlette": "starlette", "starlette": "starlette",
"tensorflow-cpu": "tensorflow-cpu>=2.3", "tensorflow-cpu": "tensorflow-cpu>=2.4,<2.11",
"tensorflow": "tensorflow>=2.4,<2.11", "tensorflow": "tensorflow>=2.4,<2.11",
"tensorflow-text": "tensorflow-text", "tensorflow-text": "tensorflow-text",
"tf2onnx": "tf2onnx", "tf2onnx": "tf2onnx",