File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -66,6 +66,8 @@ exo_pyo3_bindings = { workspace = true }
6666mlx = { git = " https://github.com/rltakashige/mlx-jaccl-fix-small-recv.git" , branch = " address-rdma-gpu-locks" }
6767mlx-lm = { git = " https://github.com/rltakashige/mlx-lm" , branch = " leo/eval-left-padding-in-batched-rotation" }
6868torch = [{ index = " pytorch-cu130" , marker = " sys_platform == 'linux'" }]
69+ torchaudio = [{ index = " pytorch-cu130" , marker = " sys_platform == 'linux'" }]
70+ torchvision = [{ index = " pytorch-cu130" , marker = " sys_platform == 'linux'" }]
6971vllm = { git = " https://github.com/hmellor/vllm.git" , branch = " transformers-v5" }
7072
7173[[tool .uv .index ]]
Original file line number Diff line number Diff line change 173173 '' ;
174174 autoPatchelfIgnoreMissingDeps = ( old . autoPatchelfIgnoreMissingDeps or [ ] ) ++ lib . optionals cudaSupport [ "libcuda.so.1" ] ;
175175 } ) ;
176+ torch-c-dlpack-ext = prev . torch-c-dlpack-ext . overrideAttrs ( old :
177+ {
178+ preFixup = ( old . preFixup or "" ) + ''addAutoPatchelfSearchPath "${ final . torch } "'' ;
179+ }
180+ ) ;
176181 xgrammar = prev . xgrammar . overrideAttrs ( old : {
177182 nativeBuildInputs = ( old . nativeBuildInputs or [ ] ) ++ [ pkgs . cmake pkgs . autoPatchelfHook ] ;
178183 } ) ;
You can’t perform that action at this time.
0 commit comments