diff --git a/requirements/cuda.txt b/requirements/cuda.txt index b5f6074d29..43b26e2b42 100644 --- a/requirements/cuda.txt +++ b/requirements/cuda.txt @@ -21,7 +21,7 @@ nvidia-cudnn-frontend>=1.13.0,<1.19.0 fastsafetensors >= 0.2.2 # QuACK and Cutlass DSL for FA4 (cute-DSL implementation) -nvidia-cutlass-dsl[cu13]>=4.4.2 +nvidia-cutlass-dsl[cu13]==4.5.0 quack-kernels>=0.3.3 # Tokenspeed_MLA for faster mla with spec decode