Pin tensorflow<=2.12.1 (#1305)

pin tensorflow
huggingface · Aug 22, 2023 · 0323172 · 0323172
1 parent 0b08a1f
commit 0323172
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",