diff --git a/crates/uv-torch/src/backend.rs b/crates/uv-torch/src/backend.rs index b9fa72141c92b..e3a5bd35f46b1 100644 --- a/crates/uv-torch/src/backend.rs +++ b/crates/uv-torch/src/backend.rs @@ -322,11 +322,16 @@ impl TorchStrategy { TorchSource::Pyx => { matches!( package_name.as_str(), - "flash-attn" + "deepspeed" + | "flash-attn" | "flash-attn-3" | "megablocks" | "natten" - | "deepspeed" + | "pyg-lib" + | "torch-cluster" + | "torch-scatter" + | "torch-sparse" + | "torch-spline-conv" | "vllm" | "torch" | "torch-model-archiver"