diff --git a/setup.py b/setup.py index 71032d361c..2ae37f238a 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",