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

giuseppe-coco · Giuseppe Coco · web-flow · commit 30b2fa7620e2 · 2024-10-29T18:36:06.000+01:00
* Add /app to PYTHONPATH * fix for issue #1415 --------- Co-authored-by: Giuseppe Coco <giuseppe.coco@besharp.it>
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
 
