diff --git a/pyproject.toml b/pyproject.toml index 40800b6db..02de11ff9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -75,6 +75,7 @@ dependencies = [ "sentence-transformers>=4.1", "setproctitle~=1.3.4", "tika-client~=0.10.0", + "torch==2.7.0+cpu; sys_platform=='linux'", "tqdm~=4.67.1", "watchdog~=6.0", "whitenoise~=6.9", @@ -172,8 +173,14 @@ psycopg-c = [ ] tiktoken = { index = "pytorch-cpu" } tokenizers = { index = "pytorch-cpu" } -torch = { index = "pytorch-cpu" } -torchvision = { index = "pytorch-cpu" } +torch = [ + { index = "pytorch-cpu", marker = "sys_platform == 'linux'" }, + { index = "pypi", marker = "sys_platform != 'linux'" }, +] +torchvision = [ + { index = "pytorch-cpu", marker = "sys_platform == 'linux'" }, + { index = "pypi", marker = "sys_platform != 'linux'" }, +] zxing-cpp = [ { url = "https://github.com/paperless-ngx/builder/releases/download/zxing-2.3.0/zxing_cpp-2.3.0-cp312-cp312-linux_x86_64.whl", marker = "sys_platform == 'linux' and platform_machine == 'x86_64' and python_version == '3.12'" }, { url = "https://github.com/paperless-ngx/builder/releases/download/zxing-2.3.0/zxing_cpp-2.3.0-cp312-cp312-linux_aarch64.whl", marker = "sys_platform == 'linux' and platform_machine == 'aarch64' and python_version == '3.12'" },