From 0f2c96cabde8d9e73060de892ed8442e5c1273af Mon Sep 17 00:00:00 2001 From: Parag Ekbote Date: Sun, 22 Mar 2026 16:26:53 +0000 Subject: [PATCH] fix dep resolving for vllm extras. --- pyproject.toml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index 5b1eb704..c8041580 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -164,6 +164,8 @@ dependencies = [ vllm = [ "vllm>=0.16.0", "ray", + "nvidia-cudnn-frontend; sys_platform == 'linux'", + "nvidia-cutlass-dsl-libs-base; sys_platform == 'linux'", ] stable-fast = [ "xformers>=0.0.30",