Skip to content

Commit d2e9b33

Browse files
committed
Fix torch installation on mac
Signed-off-by: Yuval Shkolar <yshkolar@nvidia.com>
1 parent 32a4ad1 commit d2e9b33

File tree

2 files changed

+2333
-190
lines changed

2 files changed

+2333
-190
lines changed

nemo_retriever/pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -108,8 +108,8 @@ nemotron-page-elements-v3 = { index = "test-pypi" }
108108
nemotron-graphic-elements-v1 = { index = "test-pypi" }
109109
nemotron-table-structure-v1 = { index = "test-pypi" }
110110
nemotron-ocr = { index = "test-pypi" }
111-
torch = { index = "torch-cuda"}
112-
torchvision = { index ="torch-cuda"}
111+
torch = { index = "torch-cuda", marker = "sys_platform != 'darwin'" }
112+
torchvision = { index ="torch-cuda", marker = "sys_platform != 'darwin'" }
113113
vllm = [
114114
{ url = "https://github.com/vllm-project/vllm/releases/download/v0.16.0/vllm-0.16.0+cu130-cp38-abi3-manylinux_2_35_x86_64.whl", marker = "platform_machine == 'x86_64'" },
115115
{ url = "https://github.com/vllm-project/vllm/releases/download/v0.16.0/vllm-0.16.0+cu130-cp38-abi3-manylinux_2_35_aarch64.whl", marker = "platform_machine == 'aarch64'" },

0 commit comments

Comments
 (0)