Fix tensorflow upperbound macos (#1901)

* Limited transformers version for macos * Included comment for tensorflow macos upperbound * Update runtimes/huggingface/pyproject.toml Co-authored-by: Sherif Akoush <[email protected]> --------- Co-authored-by: Sherif Akoush <[email protected]>
SeldonIO · Sep 9, 2024 · a5c82d1 · a5c82d1
1 parent eea6593
commit a5c82d1
Show file tree

Hide file tree

Showing 2 changed files with 84 additions and 2 deletions.
diff --git a/runtimes/huggingface/poetry.lock b/runtimes/huggingface/poetry.lock
diff --git a/runtimes/huggingface/pyproject.toml b/runtimes/huggingface/pyproject.toml
@@ -23,8 +23,13 @@ torch = [
     {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "<2.3.0"},
     {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "<2.4.0"}
 ]
+# tensorflow 2.17.0 not compiled for MacOS for x86 architectures
+# see: https://www.tensorflow.org/install/pip#package_location
+tensorflow = [
+    {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = ">=2.12,<2.17"},
+    {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = ">=2.12,<2.18"}
+]
 transformers = ">=4.30,<5.0"
-tensorflow = ">=2.12,<2.18"
 
 [build-system]
 requires = ["poetry-core"]