From 0323172c12cd91bff6847d3c4286c529f09f012f Mon Sep 17 00:00:00 2001 From: fxmarty <9808326+fxmarty@users.noreply.github.com> Date: Tue, 22 Aug 2023 21:52:37 +0900 Subject: [PATCH] Pin tensorflow<=2.12.1 (#1305) pin tensorflow --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 71032d361c4..2ae37f238a4 100644 --- a/setup.py +++ b/setup.py @@ -59,7 +59,7 @@ ], "exporters": ["onnx", "onnxruntime", "timm"], "exporters-gpu": ["onnx", "onnxruntime-gpu", "timm"], - "exporters-tf": ["tensorflow>=2.4", "tf2onnx", "onnx", "onnxruntime", "timm", "h5py", "numpy<1.24.0"], + "exporters-tf": ["tensorflow>=2.4,<=2.12.1", "tf2onnx", "onnx", "onnxruntime", "timm", "h5py", "numpy<1.24.0"], "diffusers": ["diffusers"], "intel": "optimum-intel>=1.10.1", "openvino": "optimum-intel[openvino]>=1.10.1",