fix[#1415]: using torch 2.2.0 in macOS x86_64 (#1417)

* Add /app to PYTHONPATH * fix for issue #1415 --------- Co-authored-by: Giuseppe Coco <[email protected]>
sinaptik-ai · Oct 29, 2024 · 30b2fa7 · 30b2fa7
1 parent c4478b5
commit 30b2fa7
Show file tree

Hide file tree

Showing 3 changed files with 130 additions and 20 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -57,6 +57,10 @@ sentence-transformers = {version = "^2.3.0", optional = true}
 sqlglot = {extras = ["rs"], version = "^25.0.3"}
 pymilvus = {version = "^2.4.6", optional = true}
 milvus-model = {version = "^0.2.7", optional = true}
+torch = [
+    { version = "2.2.0", markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'" },
+    { version = "2.4.1", markers = "sys_platform != 'darwin'" }
+]
 
 [tool.poetry.group.dev]
 optional = true

diff --git a/server/Dockerfile b/server/Dockerfile
@@ -21,6 +21,8 @@ RUN curl -sSL https://install.python-poetry.org | python3 -
 # Add Poetry to PATH
 ENV PATH="/root/.local/bin:$PATH"
 
+ENV PYTHONPATH=/app
+
 # Copy the current directory contents into the container at /app
 COPY . /app