Fix tensorflow min version depending on Python version

huggingface · Jul 12, 2024 · b632082 · b632082
1 parent dfc2b1b
commit b632082
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -176,7 +176,8 @@
     "sqlalchemy",
     "s3fs>=2021.11.1",  # aligned with fsspec[http]>=2021.11.1; test only on python 3.7 for now
     "protobuf<4.0.0",  # 4.0.0 breaks compatibility with tensorflow<2.12
-    "tensorflow>=2.6.0",  # Issue installing 2.16.0 with Python 3.8; we rely on other dependencies pinning numpy < 2
+    "tensorflow>=2.6.0; python_version<'3.10'",
+    "tensorflow>=2.16.0; python_version>='3.10'",  # Pins numpy < 2
     "tiktoken",
     "torch>=2.0.0",
     "soundfile>=0.12.1",