Merge pull request #3 from praekeltfoundation/initial-commits

Torch cpu only
praekeltfoundation · Nov 25, 2024 · 716c5bb · 716c5bb
2 parents e7eb323 + bd9e815
commit 716c5bb
Show file tree

Hide file tree

Showing 2 changed files with 60 additions and 73 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,9 +17,11 @@ prometheus-flask-exporter = "^0.23.1"
 flask-basicauth = "^0.2.0"
 sentry-sdk = {extras = ["flask"], version = "^2.18.0"}
 numpy = "1.23.5"
-torch = "2.2.2"
-sentence-transformers = "3.1.1"
-
+torch = [
+  {url = "https://download.pytorch.org/whl/cpu/torch-2.2.2%2Bcpu-cp311-cp311-linux_x86_64.whl", markers = "sys_platform == 'linux'"},
+  {url = "https://download.pytorch.org/whl/cpu/torch-2.2.2-cp311-none-macosx_10_9_x86_64.whl", markers = "sys_platform == 'darwin'"},
+]
+sentence-transformers = "^3.1.1"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.7.4"